lightning/pytorch_lightning/trainer
ananthsub 851f9e3997
Move NaN/Inf detection to a separate utilities file (#6834)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2021-04-09 01:47:02 +02:00
..
connectors Remove hardcoding of rank_zero_only.rank in accelerator connector (#6878) 2021-04-08 12:56:59 +05:30
__init__.py added trainer api docs (#4569) 2020-11-07 14:18:45 -05:00
callback_hook.py [Model Parallel] Add configure sharded model hook (#6679) 2021-03-29 14:50:51 -06:00
configuration_validator.py Add trainer.predict config validation (#6543) 2021-03-21 21:07:54 +00:00
data_loading.py Follow E231 [flake8] (#6110) 2021-03-24 12:50:50 +01:00
deprecated_api.py Simplify deprecations (#6620) 2021-03-25 15:26:38 +01:00
evaluation_loop.py Remove logger_connector legacy code (#6733) 2021-03-30 12:33:33 +02:00
ignored_warnings.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
logging.py Remove legacy support for the magic `log`/`progress_bar` keys in dict returns (#6734) 2021-03-31 00:28:04 +02:00
model_hooks.py Pass {fit,validate,test,predict} to setup() and teardown() (#6386) 2021-03-08 15:27:07 +01:00
optimizers.py Fix trainer not resetting lightning_optimizers (#6372) 2021-03-08 09:58:03 +08:00
predict_loop.py [Fix] Ensure we set the eval/train flag correctly on accelerator model (#6877) 2021-04-08 14:04:26 -04:00
properties.py Simple reproducibility with minimum boilerplate CLI training with `LightningCLI` (#4492) 2021-04-06 14:19:11 +01:00
states.py Add Trainer.validate(…) method to run one validation epoch (#4948) 2021-03-11 03:46:37 +01:00
supporters.py Remove legacy `Result` parameters (#6016) 2021-03-28 11:55:08 +02:00
trainer.py [Fix] Ensure we set the eval/train flag correctly on accelerator model (#6877) 2021-04-08 14:04:26 -04:00
training_loop.py Move NaN/Inf detection to a separate utilities file (#6834) 2021-04-09 01:47:02 +02:00
training_tricks.py Move NaN/Inf detection to a separate utilities file (#6834) 2021-04-09 01:47:02 +02:00