f4a0069ce3
* set_epoch for prediction and evaluation * minor fix in the test, warning msg was changed Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com> |
||
---|---|---|
.. | ||
batch | ||
epoch | ||
optimization | ||
__init__.py | ||
test_all.py | ||
test_evaluation_loop.py | ||
test_evaluation_loop_flow.py | ||
test_flow_warnings.py | ||
test_loop_state_dict.py | ||
test_loops.py | ||
test_training_loop.py | ||
test_training_loop_flow_dict.py | ||
test_training_loop_flow_scalar.py | ||
test_utilities.py |