Skip to content
Snippets Groups Projects
  1. Jun 29, 2022
  2. Jun 28, 2022
  3. Jun 27, 2022
  4. Jun 24, 2022
    • Yuan Liu's avatar
      [Feature]: Set different seed to different rank (#298) · 03d5c17b
      Yuan Liu authored
      * [Feature]: Set different seed for diff rank
      
      * [Feature]: Add log
      
      * [Fix]: Fix lint
      
      * [Fix]: Fix docstring
      
      * [Fix]: Fix sampler seed
      
      * [Fix]: Fix log bug
      
      * [Fix]: Change diff_seed to diff_rank_seed
      
      * [Fix]: Fix lint
      03d5c17b
  5. Jun 23, 2022
  6. Jun 22, 2022
  7. Jun 21, 2022
    • Alex Yang's avatar
      [Refactor] Refactor `after_val_epoch` to make it output metric by epoch (#278) · dceef1f6
      Alex Yang authored
      * [Refactor]:Refactor `after_val_epoch` to make it output metric by epoch
      
      * add an option for user to choose the way of outputing metric
      
      * rename variable
      
      * reformat docstring
      
      * add type alias
      
      * reformat code
      
      * add test function
      
      * add comment and test code
      
      * add comment and test code
      dceef1f6
    • Alex Yang's avatar
      [Feat] Support FSDP Training (#304) · ef946404
      Alex Yang authored
      * [Feat] Support FSDP Training
      
      * fix version comparison
      
      * change param format and move `FSDP_WRAP_POLICY` to wrapper file
      
      * add docstring and type hint,reformat code
      
      * fix type hint
      
      * fix typo, reformat code
      ef946404
  8. Jun 17, 2022
  9. Jun 16, 2022
  10. Jun 14, 2022
  11. Jun 13, 2022
    • Mashiro's avatar
      [Refactor] Refactor the accumulate gradient implemention of OptimWrapper (#284) · b7866021
      Mashiro authored
      * merge context
      
      * update unit test
      
      * add docstring
      
      * fix bug in AmpOptimWrapper
      
      * add docstring for backward
      
      * add warning and docstring for accumuate gradient
      
      * fix docstring
      
      * fix docstring
      
      * add params_group method
      
      * fix as comment
      
      * fix as comment
      
      * make default_value of loss_scale to dynamic
      
      * Fix docstring
      
      * decouple should update and should no sync
      
      * rename attribute in OptimWrapper
      
      * fix docstring
      
      * fix comment
      
      * fix comment
      
      * fix as comment
      
      * fix as comment and add unit test
      b7866021
    • Miao Zheng's avatar
      [Features]Add OneCycleLR (#296) · fd295741
      Miao Zheng authored
      * [Features]Add OnecycleLR
      
      * [Features]Add OnecycleLR
      
      * yapf disable
      
      * build_iter_from_epoch
      
      * add epoch
      
      * fix args
      
      * fix according to comments;
      
      * lr-param
      
      * fix according to comments
      
      * defaults -> default to
      
      * remove epoch and steps per step
      
      * variabel names
      fd295741
    • Mashiro's avatar
      [Fix] fix build train_loop during test (#295) · 8b0c9c5f
      Mashiro authored
      * fix build train_loop during test
      
      * fix build train_loop during test
      
      * fix build train_loop during test
      
      * fix build train_loop during test
      
      * Fix as comment
      8b0c9c5f
    • RangiLyu's avatar
      [Fix] Fix image dtype when enable_normalize=False. (#301) · 819e10c2
      RangiLyu authored
      * [Fix] Fix image dtype when enable_normalize=False.
      
      * update ut
      
      * move to collate
      
      * update ut
      819e10c2
    • Mashiro's avatar
      [Feature] Add ModuleList Sequential and ModuleDict (#299) · bcab8132
      Mashiro authored
      * add module list
      
      * add module list
      
      * fix docstring
      bcab8132
  12. Jun 10, 2022
  13. Jun 09, 2022
  14. Jun 08, 2022
  15. Jun 07, 2022
  16. Jun 06, 2022
Loading