c5248b17b7
[Enhance] Adapt test cases on Ascend NPU. ( #1728 )
2023-07-28 13:39:38 +08:00
47e033c466
[Fix] fix pbn bug ( #1466 )
2023-04-07 15:38:04 +08:00
3472ee5d2c
[Feature] Implememnt the universal visualizer for multiple task. ( #1404 )
...
* [Feature] Implememnt the universal visualizer for multiple task.
* Update tools
* Improve according to comments.
* Fix tools docs
* Add --test-cfg option and set default collate function.
2023-03-09 11:36:54 +08:00
08dc8c75d3
[Refactor] Add selfsup algorithms. ( #1389 )
...
* remove basehead
* add moco series
* add byol simclr simsiam
* add ut
* update configs
* add simsiam hook
* add and refactor beit
* update ut
* add cae
* update extract_feat
* refactor cae
* add mae
* refactor data preprocessor
* update heads
* add maskfeat
* add milan
* add simmim
* add mixmim
* fix lint
* fix ut
* fix lint
* add eva
* add densecl
* add barlowtwins
* add swav
* fix lint
* update readtherdocs rst
* update docs
* update
* Decrease UT memory usage
* Fix docstring
* update DALLEEncoder
* Update model docs
* refactor dalle encoder
* update docstring
* fix ut
* fix config error
* add val_cfg and test_cfg
* refactor clip generator
* fix lint
* pass check
* fix ut
* add lars
* update type of BEiT in configs
* Use MMEngine style momentum in EMA.
* apply mmpretrain solarize
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-06 16:53:15 +08:00
36bea13fca
[Refactor] Refactor ClsDatasample to a union DataSample. ( #1371 )
...
* [Refactor] Refactor ClsDatasample to a union DataSample.
* Add method
* Fix docstring
* Update docstring.
2023-02-23 10:07:53 +08:00
0979e78573
Rename the package name to `mmpretrain`.
2023-02-17 15:20:55 +08:00
97c4ae8805
[Improve] Update registries of mmcls. ( #1306 )
...
* [Improve] Update registries of mmcls.
* Update according to comments
2023-01-11 15:20:51 +08:00
7b9a1010f5
[Enhance] Support evaluate on both EMA and non-EMA models. ( #1204 )
...
* [Enhance] Support evaluate on both EMA and original models.
* Fix lint
2022-12-05 14:16:12 +08:00
b0007812d6
[Enhance] Enhance ArcFaceClsHead. ( #1181 )
...
* update arcface
* fix unit tests
* add adv-margins
add adv-margins
update arcface
* rebase
* update doc and fix ut
* rebase
* update code
* rebase
* use label data
* update set-margins
* Modify Arcface related method names.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 18:10:39 +08:00
940a06f645
[Refactor] Refactor to use new fileio API in MMEngine. ( #1176 )
...
* [Refactor] Refactor to use new fileio API in MMEngine.
* Add comment about why use `backend`
2022-11-21 10:56:35 +08:00
c3c1cb93aa
[Feature] Add Switch Recipe Hook. ( #1101 )
...
* add switch hook and UTs
* update doc
* update doc
* fix lint
* fix ci
* fix ci
* fix typo
* fix ci
* switchTrainAugHook to switchRecipeHook
* fix lint
* Refactor the `SwitchRecipeHook`.
* Fix windows CI
* Fix windows CI
* Fix windows CI.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-18 18:12:03 +08:00
693596bc2f
[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. ( #1055 )
...
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* update retriever
* fix lint
* add hook unit test
* Use `register_buffer` to save prototype vectors and add a progress bar
during preparing prototype.
* update UTs
* update UTs
* fix typo
* modify the hook
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-02 17:43:56 +08:00
61e9d890a6
[CI] Update circle-ci and github workflow. ( #1018 )
...
* Add deploy workflow.
* [CI] Update circle-ci and github workflow.
* Fix windows CI
* Update unit tests to save memory
2022-09-13 11:46:15 +08:00
2b88df4484
[Refactor] Refactor the data flow. ( #989 )
...
* [Refactor] Refactor the data flow.
* Add comments about data preprocessor.
* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
b8b31e9343
[Improve] Multiple fixes. ( #957 )
...
* Rename some files in unit tests.
* Rename `pipelines` folder to `transforms`.
* Fix version requirement range.
* Remove `init_random_seed` and `set_random_seed`.
* Fix unit tests
* Rename `convert_models` to `model_converters`.
2022-08-09 15:58:08 +08:00
624d22b441
[Refactor] Refactor Precise BN Hook. ( #913 )
2022-08-03 15:24:19 +08:00
43e60ad5a6
[Improve] Rename `mmcls.data` to `mmcls.structures`. ( #941 )
2022-07-29 14:18:13 +08:00
1c82863449
fix lint
2022-07-18 11:11:13 +08:00
6ed3860ea6
refactor UTs
2022-07-18 11:11:13 +08:00
Yinlei Sun
Ezra-Yu
Ma Zerun
Yixiao Fang
zzc98
Ezra-Yu