lightning/pytorch_lightning/trainer
Joost van Doorn abe795e285
Fix _module_available to detect horovod.torch properly (#12377)
Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Jirka <jirka.borovec@seznam.cz>
2022-03-28 16:00:45 +00:00
..
connectors Fix _module_available to detect horovod.torch properly (#12377) 2022-03-28 16:00:45 +00:00
__init__.py CI: precommit - docformatter (#8584) 2021-09-06 12:49:09 +00:00
callback_hook.py Fix docstrings of on_fit_start/end #12016 2022-02-21 20:50:10 +01:00
configuration_validator.py [3/3] Update lightning callbacks to `Stateful`, deprecations for old `on_save/load_checkpoint` signatures (#11887) 2022-03-25 00:06:10 +00:00
data_loading.py Deprecate `TrainerDataLoadingMixin` and move logic to `DataConnector` (#11282) 2022-01-05 21:23:57 +01:00
optimizers.py Move the `lightning_optimizers` ownership to the `Strategy` (#11444) 2022-01-18 12:58:56 +01:00
progress.py Reset the progress tracking state after sanity checking (#11218) 2021-12-23 16:36:03 +00:00
states.py 1/n Add `FaultTolerantMode` (#10645) 2021-11-22 14:58:23 +00:00
supporters.py Create the loss accumulator directly on the device (#12430) 2022-03-25 12:46:17 +01:00
trainer.py Support loading a checkpoint with QAT (#11346) 2022-03-28 16:58:33 +02:00