lightning/pytorch_lightning/trainer
Leonard Lausen 597b309f2e
Fix `Trainer.plugins` type declaration (#7288)
* Fix trainer.plugins type declaration

* Don't ClusterEnvironment(Plugin)

* fix import error, yapf formatter

* Add test

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
2021-05-04 08:42:57 +02:00
..
connectors Update Accelerator Connector for Registry (#7214) 2021-05-03 21:03:21 +00:00
__init__.py added trainer api docs (#4569) 2020-11-07 14:18:45 -05:00
callback_hook.py [bugfix] Add reloading support using BaseFinetuning (#7253) 2021-04-30 11:14:43 -04:00
configuration_validator.py Code cleaning in preparation for #7258 [3/n] (#7262) 2021-04-29 14:40:51 +02:00
data_loading.py Remove model.trainer call inside of dataloading mixin (#7317) 2021-05-03 13:53:54 -07:00
deprecated_api.py Simplify deprecations (#6620) 2021-03-25 15:26:38 +01:00
evaluation_loop.py Create pytorch_lightning/utilities/types.py (#7048) 2021-04-19 14:43:16 +02:00
ignored_warnings.py formatting 4/n: Trainer (#5720) 2021-02-03 09:25:42 +00:00
logging.py Move metrics_to_scalars to a dedicated utilities file (#7180) 2021-04-24 10:25:33 +01:00
model_hooks.py Pass {fit,validate,test,predict} to setup() and teardown() (#6386) 2021-03-08 15:27:07 +01:00
optimizers.py Enforce Lightning module as source of truth for automatic optimization (#7130) 2021-04-26 05:36:26 +00:00
predict_loop.py Code cleaning in preparation for #7258 [3/n] (#7262) 2021-04-29 14:40:51 +02:00
properties.py [feat] Add BasePredictionWriter 3/3 (#7127) 2021-04-27 20:23:55 +00:00
states.py Add Trainer.validate(…) method to run one validation epoch (#4948) 2021-03-11 03:46:37 +01:00
supporters.py fix case where an IterableDataset doesn't produce a batch for an epoch (#7294) 2021-04-30 12:45:55 +00:00
trainer.py Fix `Trainer.plugins` type declaration (#7288) 2021-05-04 08:42:57 +02:00
training_loop.py [bugfix] Fix dataloading for iterable datasets and limit_train_batches (#7306) 2021-05-03 19:50:26 +01:00
training_tricks.py Move NaN/Inf detection to a separate utilities file (#6834) 2021-04-09 01:47:02 +02:00