lightning/pytorch_lightning/core
Carlos Mocholí efd272a3ca
Pass {fit,validate,test,predict} to setup() and teardown() (#6386)
2021-03-08 15:27:07 +01:00
..
__init__.py CI: add flake8 (#4239) 2020-10-19 21:20:17 +01:00
datamodule.py Pass {fit,validate,test,predict} to setup() and teardown() (#6386) 2021-03-08 15:27:07 +01:00
decorators.py Add before_batch_transfer and after_batch_transfer hooks (#3671) 2021-02-18 06:58:12 -05:00
grads.py formatting 5/n: Core (#5721) 2021-02-08 14:29:43 -05:00
hooks.py Pass {fit,validate,test,predict} to setup() and teardown() (#6386) 2021-03-08 15:27:07 +01:00
lightning.py Refactor RunningStage usage in advance of implementing Trainer.validate() (#4945) 2021-03-06 12:40:19 +00:00
memory.py Add before_batch_transfer and after_batch_transfer hooks (#3671) 2021-02-18 06:58:12 -05:00
optimizer.py [bugfix] Check LightningOptimizer doesn't delete optimizer hooks (#6305) 2021-03-04 20:11:59 +00:00
saving.py fix duplicate console logging bug v2 (#6275) 2021-03-02 15:17:55 +05:30
step_result.py Fix for incorrect usage of detach(), cpu(), to() (#6216) 2021-03-01 15:15:52 +00:00