Merge pull request #3059 from Neilblaze:patch-fix-v2

PiperOrigin-RevId: 527379796

Author: Flax Authors

docs/notebooks/full_eval.ipynb

@@ -299,7 +299,7 @@
     "  total += len(batch['label'])\n",
     "  correct += (batch['label'] == preds.argmax(axis=1)).sum()\n",
     "\n",
-    "correc = correct.item()\n",
+    "correct = correct.item()\n",
     "correct, total, correct / total"
    ]
   },
@@ -351,7 +351,7 @@
     "  total += len(batch['label'].flatten())\n",
     "  correct += (batch['label'] == preds.argmax(axis=-1)).sum()\n",
     "\n",
-    "correc = correct.item()\n",
+    "correct = correct.item()\n",
     "correct, total, correct / total"
    ]
   },

docs/notebooks/full_eval.md

@@ -140,7 +140,7 @@ for batch in ds.as_numpy_iterator():
   total += len(batch['label'])
   correct += (batch['label'] == preds.argmax(axis=1)).sum()
 
-correc = correct.item()
+correct = correct.item()
 correct, total, correct / total
 ```
 
@@ -169,7 +169,7 @@ for batch in ds.as_numpy_iterator():
   total += len(batch['label'].flatten())
   correct += (batch['label'] == preds.argmax(axis=-1)).sum()
 
-correc = correct.item()
+correct = correct.item()
 correct, total, correct / total
 ```
 

docs/notebooks/linen_intro.ipynb

@@ -92,7 +92,7 @@
    },
    "source": [
     "Let's instantiate a `Dense` layer.\n",
-    " - Modules are actually objects in this API, so we provide _contructor arguments_ when initializing the Module.  In this case, we only have to provide the output `features` dimension."
+    " - Modules are actually objects in this API, so we provide _constructor arguments_ when initializing the Module.  In this case, we only have to provide the output `features` dimension."
    ]
   },
   {

docs/notebooks/linen_intro.md

@@ -64,7 +64,7 @@ from flax import linen as nn
 +++ {"id": "nrVbFrh1ffve"}
 
 Let's instantiate a `Dense` layer.
- - Modules are actually objects in this API, so we provide _contructor arguments_ when initializing the Module.  In this case, we only have to provide the output `features` dimension.
+ - Modules are actually objects in this API, so we provide _constructor arguments_ when initializing the Module.  In this case, we only have to provide the output `features` dimension.
 
 ```{code-cell}
 :id: EcDH20Uufc-v

examples/imagenet/imagenet.ipynb

@@ -205,7 +205,7 @@
     "#@markdown\n",
     "#@markdown **If you select no** below, then the files will be stored on the\n",
     "#@markdown *ephemeral* Colab VM. **After some time of inactivity, this VM will\n",
-    "#@markdown be restarted an any changes are lost**.\n",
+    "#@markdown be restarted and any changes are lost**.\n",
     "#@markdown\n",
     "#@markdown **If you select yes** below, then you will be asked for your\n",
     "#@markdown credentials to mount your personal Google Drive. In this case, all\n",
@@ -254,7 +254,7 @@
    },
    "outputs": [],
    "source": [
-    "# Note : In Colab, above cell changed the working direcoty.\n",
+    "# Note : In Colab, above cell changed the working directory.\n",
     "!pwd"
    ]
   },
@@ -768,7 +768,7 @@
     "if 'google.colab' in str(get_ipython()):\n",
     "  #@markdown You can upload the training results directly to https://tensorbaord.dev\n",
     "  #@markdown\n",
-    "  #@markdown Note that everbody with the link will be able to see the data.\n",
+    "  #@markdown Note that everybody with the link will be able to see the data.\n",
     "  upload_data = 'no' #@param ['yes', 'no']\n",
     "  if upload_data == 'yes':\n",
     "    !tensorboard dev upload --one_shot --logdir ./models --name 'Flax examples/mnist'"
@@ -978,7 +978,7 @@
     }
    ],
    "source": [
-    "# Define parallelized inference function in separate cell so the the cached\n",
+    "# Define parallelized inference function in separate cell so the cached\n",
     "# compilation can be used if below cell is executed multiple times.\n",
     "@jax.pmap\n",
     "def p_get_logits(images):\n",

examples/seq2seq/seq2seq.ipynb

@@ -186,7 +186,7 @@
     "#@markdown\n",
     "#@markdown **If you select no** below, then the files will be stored on the\n",
     "#@markdown *ephemeral* Colab VM. **After some time of inactivity, this VM will\n",
-    "#@markdown be restarted an any changes are lost**.\n",
+    "#@markdown be restarted and any changes are lost**.\n",
     "#@markdown\n",
     "#@markdown **If you select yes** below, then you will be asked for your\n",
     "#@markdown credentials to mount your personal Google Drive. In this case, all\n",
@@ -647,7 +647,7 @@
     "if 'google.colab' in str(get_ipython()):\n",
     "  #@markdown You can upload the training results directly to https://tensorboard.dev\n",
     "  #@markdown\n",
-    "  #@markdown Note that everbody with the link will be able to see the data.\n",
+    "  #@markdown Note that everybody with the link will be able to see the data.\n",
     "  upload_data = 'yes' #@param ['yes', 'no']\n",
     "  if upload_data == 'yes':\n",
     "    !tensorboard dev upload --one_shot --logdir ./workdirs --name 'Flax examples/seq2seq (Colab)'"