Skip to content
Snippets Groups Projects
  1. Jun 16, 2022
  2. Jun 15, 2022
  3. Jun 14, 2022
  4. 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
  5. Jun 10, 2022
  6. Jun 09, 2022
  7. Jun 08, 2022
  8. Jun 07, 2022
  9. Jun 06, 2022
  10. Jun 05, 2022
  11. Jun 01, 2022
    • Mashiro's avatar
      [Feature] Add optimizer wrapper (#265) · 3e3866c1
      Mashiro authored
      
      * Support multiple optimizers
      
      * minor refinement
      
      * improve unit tests
      
      * minor fix
      
      * Update unit tests for resuming or saving ckpt for multiple optimizers
      
      * refine docstring
      
      * refine docstring
      
      * fix typo
      
      * update docstring
      
      * refactor the logic to build multiple optimizers
      
      * resolve comments
      
      * ParamSchedulers spports multiple optimizers
      
      * add optimizer_wrapper
      
      * fix comment and docstirng
      
      * fix unit test
      
      * add unit test
      
      * refine docstring
      
      * RuntimeInfoHook supports printing multi learning rates
      
      * resolve comments
      
      * add optimizer_wrapper
      
      * fix mypy
      
      * fix lint
      
      * fix OptimizerWrapperDict docstring and add unit test
      
      * rename OptimizerWrapper to OptimWrapper, OptimWrapperDict inherit OptimWrapper, and fix as comment
      
      * Fix AmpOptimizerWrapper
      
      * rename build_optmizer_wrapper to build_optim_wrapper
      
      * refine optimizer wrapper
      
      * fix AmpOptimWrapper.step, docstring
      
      * resolve confict
      
      * rename DefaultOptimConstructor
      
      * fix as comment
      
      * rename clig grad auguments
      
      * refactor optim_wrapper config
      
      * fix docstring of DefaultOptimWrapperConstructor
      
      fix docstring of DefaultOptimWrapperConstructor
      
      * add get_lr method to OptimWrapper and OptimWrapperDict
      
      * skip some amp unit test
      
      * fix unit test
      
      * fix get_lr, get_momentum docstring
      
      * refactor get_lr, get_momentum, fix as comment
      
      * fix error message
      
      Co-authored-by: default avatarzhouzaida <zhouzaida@163.com>
      3e3866c1
  12. May 31, 2022
  13. May 30, 2022
  14. May 27, 2022
Loading