Skip to content
Snippets Groups Projects
  1. Apr 03, 2023
  2. Mar 30, 2023
  3. Mar 29, 2023
  4. Mar 28, 2023
  5. Mar 26, 2023
  6. Mar 19, 2023
  7. Mar 17, 2023
  8. Mar 16, 2023
  9. Mar 14, 2023
  10. Mar 13, 2023
  11. 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
  12. Mar 10, 2023
  13. Mar 09, 2023
  14. Mar 08, 2023
  15. Mar 07, 2023
  16. Mar 06, 2023
  17. Mar 05, 2023
  18. Mar 02, 2023
  19. 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
    • Infinity_lee's avatar
      [Docs] Translate "Debug Tricks" to English (#953) · 8a407ca2
      Infinity_lee authored
      * add english docs
      
      * fix superlink
      
      * fix
      
      * fix
Loading