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 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
  4. Feb 21, 2023
  5. Feb 14, 2023
  6. Feb 13, 2023
  7. Feb 07, 2023
Loading