lightning/tests/tests_pytorch/loops
Carlos Mocholí 4eab1f3ef1
Call `_cuda_clearCublasWorkspaces` on teardown (#16907)
2023-03-06 17:52:02 +01:00
..
optimization Remove the `*_step_end` hooks (#16791) 2023-02-20 13:04:40 +00:00
__init__.py
test_all.py Make dataloader_idx optional for batch start/end hooks (#16753) 2023-02-14 18:31:09 +01:00
test_evaluation_loop.py Call `_cuda_clearCublasWorkspaces` on teardown (#16907) 2023-03-06 17:52:02 +01:00
test_evaluation_loop_flow.py Remove the `*_step_end` hooks (#16791) 2023-02-20 13:04:40 +00:00
test_fetchers.py Move the `CombinedLoader` to an utility file (#16819) 2023-02-20 18:06:35 +01:00
test_flow_warnings.py Remove memory-retaining epoch-end hooks (#16520) 2023-02-06 17:00:36 +00:00
test_loop_state_dict.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00
test_loops.py Sequential `CombinedLoader` to flatten the eval and predict loops (#16726) 2023-02-17 17:37:11 +01:00
test_prediction_loop.py Check for dataloader_idx presence in the hooks (#16837) 2023-02-27 17:46:46 +01:00
test_progress.py copyright Lightning AI team (#16647) 2023-02-06 15:26:51 +01:00
test_training_epoch_loop.py Move `max_batches` definition to the Loops (#16820) 2023-02-22 14:01:34 +01:00
test_training_loop.py Move `max_batches` definition to the Loops (#16820) 2023-02-22 14:01:34 +01:00
test_training_loop_flow_dict.py Remove the `*_step_end` hooks (#16791) 2023-02-20 13:04:40 +00:00
test_training_loop_flow_scalar.py Remove the `*_step_end` hooks (#16791) 2023-02-20 13:04:40 +00:00