lightning/pytorch_lightning/loops/batch
Ivan Švogor 25b771ca08
Create the loss accumulator directly on the device (#12430)
Co-authored-by: Ivan Svogor <ivan.svogor@iarai.ac.at>
Co-authored-by: Carlos Mocholi <carlossmocholi@gmail.com>
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
2022-03-25 12:46:17 +01:00
..
__init__.py Loop and test restructuring (#9383) 2021-09-10 13:18:24 +00:00
training_batch_loop.py Create the loss accumulator directly on the device (#12430) 2022-03-25 12:46:17 +01:00