Skip to content
Snippets Groups Projects
  1. Mar 30, 2023
  2. Mar 29, 2023
  3. Mar 28, 2023
  4. Mar 19, 2023
  5. Mar 17, 2023
  6. Mar 16, 2023
  7. Mar 14, 2023
  8. Mar 13, 2023
  9. 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
  10. Mar 10, 2023
  11. Mar 09, 2023
  12. Mar 08, 2023
  13. Mar 06, 2023
  14. Mar 01, 2023
    • Qian Zhao's avatar
      [Feature] Enable bf16 in AmpOptimWrapper (#960) · 2ed8e343
      Qian Zhao authored
      * support bf16 in AmpOptimWrapper
      
      * add docstring
      
      * modify docs
      
      * add unittests for bf16 in AmpOptimWrapper
      
      * fix type
      
      * fix to pass ci
      
      * fix ut skip logic to pass ci
      
      * fix as comment
      
      * add type hints
      
      * fix docstring and add warning information
      
      * remove check for pytorch>=1.6 in unittest
      
      * modify unittest
      
      * modify unittest
      
      * remove torch.float32 && torch.float64 from valid dtypes
      
      * fix as comments
      
      * minor refine docstring
      
      * fix unittest parameterized to pass CI
      
      * fix unittest && add back torch.float32, torch.float64
      2ed8e343
  15. Feb 27, 2023
  16. Feb 24, 2023
  17. Feb 23, 2023
  18. Feb 22, 2023
  19. Feb 21, 2023
  20. Feb 20, 2023
Loading