- Nov 24, 2022
-
-
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
-
- Nov 17, 2022
-
-
RangiLyu authored
-
- Nov 02, 2022
-
-
Mashiro authored
* [Fix] Change the default value of map_location in PretrainedInit to cpu * Fix as comment
-
- Oct 28, 2022
-
-
Mashiro authored
* [Fix] Fix cound not handle string data * Minor refine * Refine type hint Refine type hint * fix as comment * Minor refine * Update mmengine/model/base_model/data_preprocessor.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
-
Mashiro authored
* [Fix] Fix is_model_wrapper * [Fix] Fix is_model_wrapper
-
- Oct 24, 2022
-
-
wangjiangben-hw authored
* init npu * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/dist/dist.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * change to is_hccl_backend * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add comment with AmpOptimWrapper * Update mmengine/runner/amp.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/runner/amp.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * add npu fn in base_model * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * clean lint * Update mmengine/optim/optimizer/amp_optimizer_wrapper.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/model/base_model/base_model.py Co-authored-by:
Mashiro <57566630+HAOCHENYE@users.noreply.github.com> * add is_npu_available * try to fix * Add comments * Refine grammar Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by:
Mashiro <57566630+HAOCHENYE@users.noreply.github.com> Co-authored-by:
HAOCHENYE <21724054@zju.edu.cn>
-
- Oct 18, 2022
-
-
shenmishajing authored
* add non_blocking feature to BaseDataPreprocessor * Update mmengine/model/base_model/data_preprocessor.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * move new parameters to the last to avoid bc issue * Update mmengine/model/base_model/data_preprocessor.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> * Update mmengine/model/base_model/data_preprocessor.py Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com> Co-authored-by:
Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
-
- Oct 11, 2022
-
-
Mashiro authored
Co-authored-by:
ubuntu <ubuntu@localhost.localdomain>
-
- Oct 10, 2022
-
-
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
-
- Oct 09, 2022
-
-
Mashiro authored
-
- Oct 08, 2022
-
-
Younghwan Na authored
-
Mashiro authored
-
Zaida Zhou authored
-
- Sep 15, 2022
-
-
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
-
- Sep 13, 2022
-
-
Mashiro authored
-
- Sep 09, 2022
- Aug 28, 2022
-
-
Mashiro authored
* regist pytorch ddp and dp, add unit test * minor refine * Support check custom wrapper * enhance ut
-
- Aug 26, 2022
-
-
RangiLyu authored
* [Fix] Fix EMAHook trigger train loop init during testing. * fix sync buffer * update ut * fix sync buffer * fix sync buffer
-
- Aug 24, 2022
-
-
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
-
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
-
- Aug 23, 2022
-
-
Zaida Zhou authored
* [Refactor] Refactor the import rule * minor refinement * add a comment
-
- Aug 15, 2022
-
-
Mashiro authored
-
- Jul 22, 2022
-
-
RangiLyu authored
-
- Jul 08, 2022
-
-
Mashiro authored
* fix missing device ids in wrap_model * clean the code * use default broadcast_buffers * refine MMSeparateDistributedDataParallel * rename tmp variable * refine docstring * add type hints * refactor docstring of ddp model * add arg in docstring * minor refine * better ddp link
-
- Jun 28, 2022
-
-
Mashiro authored
-
Mashiro authored
* assert image shape before forward * add unit test * enhance error message * allow gray image input * fix as comment * fix unit test * fix unit test
-
Zaida Zhou authored
-
- Jun 22, 2022
-
-
Jiazhen Wang authored
* modify cuda() to to() * rollback load_checkpoint * refine runner * add TODO
-
- Jun 21, 2022
- Jun 14, 2022
-
-
Alex Yang authored
-
- Jun 13, 2022
-
-
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
-
RangiLyu authored
* [Fix] Fix image dtype when enable_normalize=False. * update ut * move to collate * update ut
-
Mashiro authored
* add module list * add module list * fix docstring
-
- Jun 09, 2022
-
-
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
-
Mashiro authored
* fix BaseDataPreprocessor * fix BaseDataPreprocessor * change device type to torch.device * change device type to torch.device * fix cpu method of base model
-
- Jun 08, 2022
- Jun 07, 2022
-
-
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
-