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
    • 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
    • Mashiro's avatar
      [Fix] Rename data_list and support loading from ceph in dataset (#240) · 538ff48a
      Mashiro authored
      * rename datalist and support load ceph
      
      * rename datalist and support load ceph
      
      * remove check disk file path in _load_metainfo
      
      * fix rename error
      
      * fix rename error
      
      * unit test error
      
      * fix rename error
      
      * remove unnecessary code
      
      * fix lint
      538ff48a
  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
  15. May 26, 2022
Loading