lightning/tests/loops
Carlos Mocholí f4505ce6b2
Construct the hook kwargs inside each loop (#12100)
Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com>
2022-05-03 17:08:02 +02:00
..
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
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 Construct the hook kwargs inside each loop (#12100) 2022-05-03 17:08:02 +02:00
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 Add `enable_model_summary` flag and deprecate `weights_summary` (#9699) 2021-10-13 17:20:54 +05:30
test_training_loop_flow_scalar.py Construct the hook kwargs inside each loop (#12100) 2022-05-03 17:08:02 +02:00
test_utilities.py Have the outputs match the loops format (#12182) 2022-03-08 18:10:18 +00:00