Skip to content
Snippets Groups Projects
  1. Apr 11, 2023
  2. Mar 12, 2023
    • Qian Zhao's avatar
      [Feature] Support torch.compile since PyTorch2.0 (#976) · 0d25625b
      Qian Zhao authored
      * 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
  3. Mar 09, 2023
  4. Mar 05, 2023
  5. Mar 01, 2023
  6. Feb 21, 2023
  7. Feb 07, 2023
Loading