Skip to content
Snippets Groups Projects
  1. Apr 05, 2023
  2. Apr 04, 2023
  3. Apr 03, 2023
  4. Mar 30, 2023
  5. Mar 29, 2023
  6. Mar 28, 2023
  7. Mar 26, 2023
  8. Mar 19, 2023
  9. Mar 17, 2023
  10. Mar 16, 2023
  11. Mar 14, 2023
  12. Mar 13, 2023
  13. 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
      0d25625b
  14. Mar 10, 2023
  15. Mar 09, 2023
  16. Mar 08, 2023
  17. Mar 07, 2023
  18. Mar 06, 2023
  19. Mar 05, 2023
  20. Mar 02, 2023
Loading