* enable compile configurations to support torch.compile in Runner * enable compilation in train, val and test * fix as comments * add docstring to illustrate usage * minor refine error message * add unittests * fix ut skip * add logging message to inform users * compile `train_step`, `val_step`, `test_step` instead * fix as comments * revert to compile `train_step` only due to pt2 issue * add documentation about torch.compile |
||
---|---|---|
.. | ||
debug_tricks.md | ||
epoch_to_iter.md | ||
model_analysis.md | ||
resume_training.md | ||
save_gpu_memory.md | ||
set_interval.md | ||
set_random_seed.md | ||
speed_up_training.md |