Merge pull request #4524 from cool-RR:patch-5

PiperOrigin-RevId: 725221248
google · Feb 10, 2025 · d935e53 · d935e53
2 parents 7efdfc8 + cc4279c
commit d935e53
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/guides/training_techniques/batch_norm.rst b/docs/guides/training_techniques/batch_norm.rst
@@ -141,8 +141,8 @@ collection.
 Modifying ``flax.linen.apply``
 ******************************
 
-When using :meth:`flax.linen.apply <flax.linen.apply>` to run your model with the ``train==True``
-argument (that is, you have ``use_running_average==False`` in the call to ``BatchNorm``), you
+When using :meth:`flax.linen.apply <flax.linen.apply>` to run your model with the ``train=True``
+argument (that is, you have ``use_running_average=False`` in the call to ``BatchNorm``), you
 need to consider the following:
 
 * ``batch_stats`` must be passed as an input variable.