docs: configure_sync_batchnorm, amp, readme python/conda badge (Lightning-AI#3328)

* fix(docs): change to configure_sync_batchnorm for sync bn hook

* Update docs/source/lightning-module.rst

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>

* distinguish pypi and conda download badges

* python version badge and original pypi download badge

* Update docs/source/lightning-module.rst

Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

Author: 3 people

README.md

@@ -18,9 +18,11 @@
 </p>
 
 
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pytorch-lightning)](https://pypi.org/project/pytorch-lightning/)
 [![PyPI Status](https://badge.fury.io/py/pytorch-lightning.svg)](https://badge.fury.io/py/pytorch-lightning)
 [![PyPI Status](https://pepy.tech/badge/pytorch-lightning)](https://pepy.tech/project/pytorch-lightning)
 [![Conda](https://img.shields.io/conda/v/conda-forge/pytorch-lightning?label=conda&color=success)](https://anaconda.org/conda-forge/pytorch-lightning)
+[![Conda](https://img.shields.io/conda/dn/conda-forge/pytorch-lightning?color=blue&label=conda%20downloads)](https://anaconda.org/conda-forge/pytorch-lightning)
 [![DockerHub](https://img.shields.io/docker/pulls/pytorchlightning/pytorch_lightning.svg)](https://hub.docker.com/r/pytorchlightning/pytorch_lightning)
 [![codecov](https://codecov.io/gh/PyTorchLightning/pytorch-lightning/branch/master/graph/badge.svg)](https://codecov.io/gh/PyTorchLightning/pytorch-lightning)
 

docs/source/lightning_module.rst

@@ -897,6 +897,12 @@ Pointer to the trainer
 
 ------------
 
+use_amp
+~~~~~~~
+True if using Automatic Mixed Precision (AMP)
+
+------------
+
 use_ddp
 ~~~~~~~
 True if using ddp
@@ -1021,7 +1027,7 @@ configure_ddp
 configure_sync_batchnorm
 ~~~~~~~~~~~~~~~~~~~~~~~~
 
-.. autofunction:: pytorch_lightning.core.lightning.LightningModule.configure_ddp
+.. autofunction:: pytorch_lightning.core.lightning.LightningModule.configure_sync_batchnorm
     :noindex:
 
 get_progress_bar_dict
@@ -1153,28 +1159,28 @@ on_pretrain_routine_end
 .. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_pretrain_routine_end
     :noindex:
 
-on_test_epoch_start
+on_test_batch_start
 ~~~~~~~~~~~~~~~~~~~
 
-.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_epoch_start
+.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_batch_start
     :noindex:
 
-on_test_epoch_end
+on_test_batch_end
 ~~~~~~~~~~~~~~~~~
 
-.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_epoch_end
+.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_batch_end
     :noindex:
 
-on_test_batch_start
+on_test_epoch_start
 ~~~~~~~~~~~~~~~~~~~
 
-.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_batch_start
+.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_epoch_start
     :noindex:
 
-on_test_batch_end
+on_test_epoch_end
 ~~~~~~~~~~~~~~~~~
 
-.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_batch_end
+.. autofunction:: pytorch_lightning.core.hooks.ModelHooks.on_test_epoch_end
     :noindex:
 
 on_train_batch_start