lightning/tests/loops
Carlos Mocholí e0f2e041b9
Share the training step output data via `ClosureResult` (#9349)
2021-09-10 11:40:20 +00:00
..
batch rewrite and improve tests for truncated back-propagation (#9369) 2021-09-08 20:32:59 +00:00
__init__.py Add `state_dict` to loops (#8197) 2021-07-01 15:54:37 +00:00
test_closure.py Share the training step output data via `ClosureResult` (#9349) 2021-09-10 11:40:20 +00:00
test_loop_state_dict.py Share the training step output data via `ClosureResult` (#9349) 2021-09-10 11:40:20 +00:00
test_loops.py Add `ManualOptimization` loop (#9266) 2021-09-08 02:26:39 +02:00
test_utilities.py Share the training step output data via `ClosureResult` (#9349) 2021-09-10 11:40:20 +00:00