| .. |
|
optimization
|
Updated check on model step output types from dict to Mapping (#18657)
|
2023-09-28 20:31:40 -04:00 |
|
__init__.py
|
…
|
|
|
test_all.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_evaluation_loop.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_evaluation_loop_flow.py
|
Always pass the correct batch index to the automatic optimization loop (#18619)
|
2023-10-03 21:23:36 +02:00 |
|
test_fetchers.py
|
Always pass the correct batch index to the automatic optimization loop (#18619)
|
2023-10-03 21:23:36 +02:00 |
|
test_flow_warnings.py
|
ruff: enable & fixing RET (#17540)
|
2023-05-05 09:34:40 +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
|
Fix zero-grad behavior when entering the validation loop (#18710)
|
2023-10-09 18:07:54 -04:00 |
|
test_prediction_loop.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_progress.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_training_epoch_loop.py
|
Always pass the correct batch index to the automatic optimization loop (#18619)
|
2023-10-03 21:23:36 +02:00 |
|
test_training_loop.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_training_loop_flow_dict.py
|
ruff: replace isort with ruff +TPU (#17684)
|
2023-09-26 11:54:55 -04:00 |
|
test_training_loop_flow_scalar.py
|
Always pass the correct batch index to the automatic optimization loop (#18619)
|
2023-10-03 21:23:36 +02:00 |
a26424e89e