Skip to content
Snippets Groups Projects
Unverified Commit 0d25625b authored by Qian Zhao's avatar Qian Zhao Committed by GitHub
Browse files

[Feature] Support torch.compile since PyTorch2.0 (#976)

* 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
parent 6ea23a2f
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment