lightning/tests/loops
rohitgr7 a68fe66705 Update core/lightning.py to core/module.py (#12740) 2022-05-18 08:03:55 +05:30
..
batch Construct the hook kwargs inside each loop (#12100) 2022-05-03 17:08:02 +02:00
epoch Construct the hook kwargs inside each loop (#12100) 2022-05-03 17:08:02 +02:00
optimization Have the outputs match the loops format (#12182) 2022-03-08 18:10:18 +00:00
__init__.py Add `state_dict` to loops (#8197) 2021-07-01 15:54:37 +00:00
test_all.py update tests for v2 (#11487) 2022-01-16 21:43:37 +05:30
test_evaluation_loop.py `set_epoch` for validation and prediction data loaders (#12197) 2022-03-28 19:27:04 +00:00
test_evaluation_loop_flow.py Update core/lightning.py to core/module.py (#12740) 2022-05-18 08:03:55 +05:30
test_flow_warnings.py Add `enable_model_summary` flag and deprecate `weights_summary` (#9699) 2021-10-13 17:20:54 +05:30
test_loop_state_dict.py Support optimizer step progress tracking with manual optimization (#11848) 2022-02-16 21:27:27 +00:00
test_loops.py Drop PyTorch 1.7 support (#12432) 2022-03-27 21:31:20 +00:00
test_training_loop.py Integrate global step with progress tracking (#11805) 2022-03-07 19:21:37 +00:00
test_training_loop_flow_dict.py Update core/lightning.py to core/module.py (#12740) 2022-05-18 08:03:55 +05:30
test_training_loop_flow_scalar.py Update core/lightning.py to core/module.py (#12740) 2022-05-18 08:03:55 +05:30
test_utilities.py Have the outputs match the loops format (#12182) 2022-03-08 18:10:18 +00:00