Skip to content
Snippets Groups Projects
  1. Jul 05, 2022
  2. 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
  3. Jun 23, 2022
  4. Jun 22, 2022
  5. Jun 21, 2022
  6. Jun 17, 2022
  7. Jun 16, 2022
  8. Jun 15, 2022
  9. 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
  10. Jun 10, 2022
  11. Jun 09, 2022
  12. Jun 07, 2022
    • Mashiro's avatar
      [Feature]: add base model, ddp model wrapper and unit test (#268) · f04fec73
      Mashiro authored
      * add base model, ddp model and unit test
      
      * add unit test
      
      * fix unit test
      
      * fix docstring
      
      * fix cpu unit test
      
      * refine base data preprocessor
      
      * refine base data preprocessor
      
      * refine interface of ddp module
      
      * remove optimizer hook
      
      * add forward
      
      * fix as comment
      
      * fix unit test
      
      * fix as comment
      
      * fix build optimizer wrapper
      
      * rebase main and fix unit test
      
      * stack_batch support stacking ndim tensor, add docstring for merge dict
      
      * fix lint
      
      * fix test loop
      
      * make precision_context effective to data_preprocessor
      
      * fix as comment
      
      * fix as comment
      
      * refine docstring
      
      * change collate_data output typehints
      
      * rename to_rgb to bgr_to_rgb and rgb_to_bgr
      
      * support build basemodel with built DataPreprocessor
      
      * fix as comment
      
      * fix docstring
      f04fec73
    • RangiLyu's avatar
      [Enhance] Enhance checkpoint meta info. (#279) · ad965a53
      RangiLyu authored
      ad965a53
  13. Jun 06, 2022
  14. Jun 05, 2022
  15. 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
  16. May 31, 2022
    • Zaida Zhou's avatar
      [Feature] Support multiple optimizers (#235) · f1da9a1d
      Zaida Zhou 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
      
      * refine docstring
      
      * RuntimeInfoHook supports printing multi learning rates
      
      * resolve comments
      
      * fix typo
      f1da9a1d
    • Jiazhen Wang's avatar
      [Enhance] Improve Exception in call_hook (#247) · f2190de7
      Jiazhen Wang authored
      * improve exception in call_hook
      
      * refine unit test
      
      * add test_call_hook
      
      * refine
      
      * update docstring and ut
      f2190de7
  17. May 27, 2022
  18. May 26, 2022
  19. May 25, 2022
  20. May 24, 2022
  21. May 21, 2022
  22. May 20, 2022
  23. May 19, 2022
  24. May 18, 2022
  25. May 17, 2022
  26. May 12, 2022
  27. May 10, 2022
Loading