lightning/pytorch_lightning/trainer
Sean Naren f0ab74dc2f
Expose scaler in amp plugin (#4737)
2020-11-18 22:30:47 +00:00
..
connectors Sharded Plugin 2/n: Allow ddp plugin to modify optimizer state saving (#4675) 2020-11-18 16:38:35 +00:00
__init__.py added trainer api docs (#4569) 2020-11-07 14:18:45 -05:00
callback_hook.py Fix setup callback hook to pass LightningModule through (#4608) 2020-11-13 19:34:46 -05:00
configuration_validator.py ref: added data connector (#3285) 2020-08-31 11:08:22 -04:00
data_loading.py Skip replacing dataloader sampler if it's already a distributed sampler (#4273) 2020-10-23 17:34:07 +01:00
evaluation_loop.py [FEAT] Refactor logging 3/3 [v1] (#4552) 2020-11-11 17:05:24 +00:00
ignored_warnings.py added copyright notices (#3062) 2020-08-19 22:03:22 -04:00
logging.py [FEAT] logging refactors 1/n (#4439) 2020-11-02 20:51:43 +00:00
model_hooks.py CI: add flake8 (#4239) 2020-10-19 21:20:17 +01:00
optimizers.py Clean up optimizer code (#3587) 2020-10-21 21:12:48 +02:00
properties.py Adds shortcut for path to log (#4573) 2020-11-08 12:16:22 -05:00
states.py Make trainer.state a read-only property (#3109) 2020-08-24 16:49:33 +02:00
supporters.py [test] Accumulated gradient optimization tests (#4477) 2020-11-02 23:44:11 +00:00
trainer.py Makes automatic optimization a model attribute (#4602) 2020-11-14 11:13:42 +06:30
training_loop.py Expose scaler in amp plugin (#4737) 2020-11-18 22:30:47 +00:00
training_tricks.py CI: add flake8 (#4239) 2020-10-19 21:20:17 +01:00