1
0
mirror of https://github.com/open-mmlab/mmpretrain.git synced 2025-06-03 06:40:40 +08:00

9 Commits

Author SHA1 Message Date
Ma Zerun
5bd088ef43
[Fix] Update torchvision transform wrapper ()
* Update torchvision transform wrapper

* Update requirements

* fix unit tests

---------

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-05-26 17:56:09 +08:00
Ezra-Yu
99e48116aa
[Feature] Register torchvision transforms into mmcls ()
* [Enhance] Add stochastic depth decay rule in resnet. ()

* add stochastic depth decay rule to drop path rate

* add default value

* update

* pass ut

* update

* pass ut

* remove np

* rebase

* update ToPIL and ToNumpy

* rebase

* rebase

* rebase

* rebase

* add readme

* fix review suggestions

* rebase

* fix conflicts

* fix conflicts

* fix lint

* remove comments

* remove useless code

* update docstring

* update doc API

* update doc

---------

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2023-04-13 18:05:57 +08:00
Ma Zerun
4f5b38f225
[Refactor] Update almost tools and add unit tests for these tools. ()
* [Refactor] Update almost tools and add unit tests for these tools.

* Fix Windows UT.
2023-03-17 10:50:51 +08:00
Ma Zerun
a05c79e806
[Refactor] Move transforms in mmselfsup to mmpretrain. ()
* [Refactor] Move transforms in mmselfsup to mmpretrain.

* Update transform docs and configs. And register some mmcv transforms in
mmpretrain.

* Fix missing transform wrapper.

* update selfsup transforms

* Fix UT

* Fix UT

* update gaussianblur inconfigs

---------

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-03 15:01:11 +08:00
mzr1996
0979e78573 Rename the package name to mmpretrain. 2023-02-17 15:20:55 +08:00
Ma Zerun
97c4ae8805
[Improve] Update registries of mmcls. ()
* [Improve] Update registries of mmcls.

* Update according to comments
2023-01-11 15:20:51 +08:00
Ma Zerun
b7d0d521eb
[Refactor] Use mmengine instead of mmcv and refactor some transforms. ()
* [Refactor] Refactor the rest data transforms and fix some docstring.

* Use `mmengine` instead of `mmcv`

* Refactor KFold dataset tools

* Fix docstring according to comments
2022-08-24 15:59:02 +08:00
Ma Zerun
20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. ()
* [CI] Add Circle CI for mmcls 1.x

* Fix circle-ci config

* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6

* Compat the behavior of LongTensor in PyTorch 1.6

* Avoid random behavior in the unit test of Lighting

* Fix cuda ci

* Remove github workflow temporarily.
2022-08-22 15:02:08 +08:00
Ma Zerun
b8b31e9343
[Improve] Multiple fixes. ()
* 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