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

PiperOrigin-RevId: 725221248

Author: Flax Authors

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.