Skip to content
Snippets Groups Projects
  1. Nov 24, 2022
    • Mashiro's avatar
      [Enhance] Enhance compatibility of `revert_sync_batchnorm` (#695) · ded73f3a
      Mashiro authored
      * [Enhance] Enhance revert_sync_batchnorm and convert_sync_batchnorm
      
      * [Enhance] Enhance revert_sync_batchnorm and convert_sync_batchnorm
      
      * Fix unit test
      
      * Add coments
      
      * Refine comments
      
      * clean the code
      
      * revert convert_sync_batchnorm
      
      * revert convert_sync_batchnorm
      
      * refine comment
      
      * fix CI
      
      * fix CI
      ded73f3a
  2. Nov 21, 2022
  3. Nov 17, 2022
  4. Oct 28, 2022
  5. Oct 18, 2022
  6. Oct 10, 2022
    • Mashiro's avatar
      [Feature] Add test time augmentation base model. (#538) · 62456217
      Mashiro authored
      * First commit
      
      * add BaseTestTimeAugModel
      
      * Add unit test
      
      * move loop logic to test_step
      
      * fix ddp test
      
      * rename model to module
      
      * optim __init__
      
      * Fix as comment
      
      * Fix as comment
      
      * make val_step should not be called
      
      * make tta do not inherit base model
      
      * Fix unit test
      
      * Enhance docstring
      
      * Fix as comment
      
      * Fix as comment
      
      * minor refine
      
      * minor refine
      
      * minor refine
      
      * fix unit test
      
      * minor refine
      
      * minor refine
      
      * minor refine
      
      * minor refine
      
      * minor refine
      
      * minor refine
      
      * fix unit test
      62456217
  7. Oct 08, 2022
  8. Sep 15, 2022
    • Mashiro's avatar
      [Feature] Support convert `BN` to `SyncBN` by config (#506) · 8ee31dbc
      Mashiro authored
      * [Feature] Support convert BN to SyncBN by config
      
      * make unit test compatible with cpu
      
      * Fix as comment
      
      * fix unit test
      
      * change signature of convert_sync_batchnorm: rename sync_bn to implemention
      
      * fix unit test
      
      * fix unit test
      8ee31dbc
  9. Sep 13, 2022
  10. Aug 28, 2022
  11. Aug 24, 2022
    • Mashiro's avatar
      [Refactor] Refactor data flow to make the interface more natural (#468) · 8770c6c7
      Mashiro authored
      * [Refactor]: modify interface of Visualizer.add_datasample (#365)
      
      * [Refactor] Refactor data flow: refine `data_preprocessor`. (#359)
      
      * refine data_preprocessor
      
      * remove unused BATCH_DATA alias
      
      * Fix type hints
      
      * rename move_data to cast_data
      
      * [Refactor] Refactor data flow: collate data in `collate_fn` of `DataLoader`  (#323)
      
      * acollate data in dataloader
      
      * fix docstring
      
      * refine comment
      
      * fix as comment
      
      * refactor default collate and psedo collate
      
      * foramt test file
      
      * fix docstring
      
      * fix as comment
      
      * rename elem to data_item
      
      * minor fix
      
      * fix as comment
      
      * [Refactor] Refactor data flow: `data_batch` argument of `Evaluator.process is a `dict` (#360)
      
      * refine evaluator and metric
      
      * compatible with new default collate
      
      * replace default collate with pseudo
      
      * Handle data_batch in metric
      
      * fix unit test
      
      * fix unit test
      
      * fix unit test
      
      * minor refine
      
      * make data_batch optional
      
      make data_batch optional
      
      * rename outputs to predictions
      
      * fix ut
      
      * rename predictions to outputs
      
      * fix docstring
      
      * fix docstring
      
      * fix unit test
      
      * make outputs and data_batch to kwargs
      
      * fix unit test
      
      * keep signature of metric
      
      * fix ut
      
      * rename pred_sample arguments to data_sample(Visualizer)
      
      * fix loop and ut
      
      * [refactor]: Refactor model dataflow (#398)
      
      * [Refactor] Refactor data flow: refine `data_preprocessor`. (#359)
      
      * refine data_preprocessor
      
      * remove unused BATCH_DATA alias
      
      * Fix type hints
      
      * rename move_data to cast_data
      
      * refactor model data flow
      
      tmp_commt
      
      tmp commit
      
      * make val_cfg and test_cfg optional
      
      * roll back runner
      
      * pass test mmdet
      
      * fix as comment
      
      fix as comment
      
      fix ci in DataPreprocessor
      
      * fix ut
      
      * fix ut
      
      * fix rebase main
      
      * [Fix]: Fix test val ddp (#462)
      
      * [Fix] Fix docstring and type hint of data flow (#463)
      
      * Fix docstring of data flow
      
      * change signature of hook
      
      * fix unit test
      
      * resolve conflicts
      
      * fix lint
      8770c6c7
    • Zaida Zhou's avatar
      [Refactor] Refactor code structure (#395) · 7e1d7af2
      Zaida Zhou authored
      * Rename data to structure
      
      * adjust the way to import module
      
      * adjust the way to import module
      
      * rename Structure to Data Structures in docs api
      
      * rename structure to structures
      
      * support using some modules of mmengine without torch
      
      * fix circleci config
      
      * fix circleci config
      
      * fix registry ut
      
      * minor fix
      
      * move init method from model/utils to model/weight_init.py
      
      * move init method from model/utils to model/weight_init.py
      
      * move sync_bn to model
      
      * move functions depending on torch to dl_utils
      
      * format import
      
      * fix logging ut
      
      * add weight init in model/__init__.py
      
      * move get_config and get_model to mmengine/hub
      
      * move log_processor.py to mmengine/runner
      
      * fix ut
      
      * Add TimeCounter in dl_utils/__init__.py
      7e1d7af2
  12. Aug 15, 2022
  13. Aug 09, 2022
  14. Jun 28, 2022
  15. Jun 22, 2022
    • Mashiro's avatar
      [Feature] Add autocast wrapper (#307) · 312f264e
      Mashiro authored
      * add autocast wrapper
      
      * fix docstring
      
      * fix docstring
      
      * fix compare version
      
      * fix unit test
      
      * fix incompatible arguments
      
      * fix as comment
      
      * fix unit test
      
      * rename auto_cast to autocast
      312f264e
  16. Jun 21, 2022
    • 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
  17. Jun 13, 2022
  18. Jun 09, 2022
    • Mashiro's avatar
      [Enhance] Enhance img data preprocessor (#290) · 931db990
      Mashiro authored
      * fix BaseDataPreprocessor
      
      * fix BaseDataPreprocessor
      
      * change device type to torch.device
      
      * change device type to torch.device
      
      * fix cpu method of base model
      
      * Allow ImgDataPreprocessor do not normalize
      
      * remove unnecessary type ignore
      
      * make mean and std optional
      
      * refine docstring
      931db990
    • Mashiro's avatar
      [Fix] Fix BaseDataPreprocessor and BaseModel (#285) · a9afdad7
      Mashiro authored
      * fix BaseDataPreprocessor
      
      * fix BaseDataPreprocessor
      
      * change device type to torch.device
      
      * change device type to torch.device
      
      * fix cpu method of base model
      a9afdad7
  19. Jun 08, 2022
  20. 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
  21. Jun 06, 2022
  22. 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
  23. May 30, 2022
  24. May 19, 2022
  25. Apr 08, 2022
  26. Feb 27, 2022
  27. Feb 26, 2022
Loading