Update training_tricks.py (#3151)
Update training_tricks.py (#3151)

* Update training_tricks.py

* pep

Co-authored-by: Rohit Gupta <cb3c4569854e7764e66b4cfd98eab60a0e5fb067@gmail.com>
Co-authored-by: mergify[bot] <b09a6ee808b67e98a221404e7aaa52e0398a4954@users.noreply.github.com>
williamFalcon/pytorch-lightning,williamFalcon/pytorch-lightning