lightning/pytorch_lightning/trainer
Sean Naren ffdcb62e8f
Make parallel devices optional across all plugins (#6051)
* Make parallel devices optional across all plugins so that they can be instantiated

* Add any to types to capture vars passed in
2021-02-18 12:09:53 +00:00
..
connectors Make parallel devices optional across all plugins (#6051) 2021-02-18 12:09:53 +00:00
__init__.py added trainer api docs (#4569) 2020-11-07 14:18:45 -05:00
callback_hook.py PoC: Accelerator refactor (#5743) 2021-02-12 15:48:56 -05:00
configuration_validator.py Add PredictLoop (#5752) 2021-02-16 17:11:56 -05:00
data_loading.py Trainer only references accelerator (#6039) 2021-02-17 21:41:18 +01:00
deprecated_api.py move accelerator_connector.py to the connectors subfolder (#6033) 2021-02-17 18:38:08 +00:00
evaluation_loop.py [BugFix] Add on_epoch_end hook after on_test/validation_epoch_end hook (#5986) 2021-02-17 18:21:49 +01:00
ignored_warnings.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
logging.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
model_hooks.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
optimizers.py [feat] Add Trainer(stochastic_weight_avg=True/False) (#6038) 2021-02-17 23:10:05 +00:00
predict_loop.py Add PredictLoop (#5752) 2021-02-16 17:11:56 -05:00
properties.py Trainer only references accelerator (#6039) 2021-02-17 21:41:18 +01:00
states.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
supporters.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
trainer.py [feat] Add Trainer(stochastic_weight_avg=True/False) (#6038) 2021-02-17 23:10:05 +00:00
training_loop.py drop deprecated result object 1/n (#5005) 2021-02-17 18:58:28 -05:00
training_tricks.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00