mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
969 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

51 Commits (4584a07e65ea747b30971c256c7636fa3455f220)

Author SHA1 Message Date
Yixiao Fang 1ee9bbe050
[Docs] Update links (#1457) 
* update links

* update readtherdocs

* update

* update

* fix lint

* update

* update

* update

* update cov branch

* update

* update

* update
 2023-04-06 20:58:52 +08:00
Ma Zerun dda3d6565b
[Docs] Update generate_readme.py and readme files. (#1388) 
* Update generate_readme.py and readme files.

* Update reamde

* Update docs

* update metafile

* update simmim readme

* update

* update mae

* fix lint

* update mocov2

* update readme pic

* fix lint

* Fix mmcls download links.

* Fix Chinese docs.

* Decrease readthedocs requirements.

---------

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
 2023-03-02 13:29:07 +08:00
Ma Zerun c48cfa9f47
[Docs] Improve the ResNet model page. (#1118) 
* [Docs] Improve the ResNet model page.

* Fix lint.

* Update stat.py
 2022-11-08 11:09:51 +08:00
Ma Zerun 29f066f7fb
[Improve] Speed up data preprocessor. (#1064) 
* [Improve] Speed up data preprocessor.

* Add ClsDataSample serialization override functions.

* Add unit tests

* Modify configs to fit new mixup args.

* Fix `num_classes` of the ImageNet-21k config.

* Update docs.
 2022-10-17 17:08:18 +08:00
mzr1996 d90dfc3d99 [Docs] Use relative link to config instead of abs link in README. 2022-09-22 09:59:06 +08:00
Ma Zerun e9e2d48cb2
[Improve] Update benchmark scripts (#1028) 
* Update train benchmark scripts

* Add `--cfg-options` for dev scripts and enhance `--range`.

* Fix bug of regex expression.

* Fix minor bugs

* Update ShuffleNet configs

* Update rsb-a1 configs and label smooth loss mode.

* Update inference dev scripts

* From `mmengine` instead of `mmcv` import fileio.

* Fix lint

* Update pre-commit hook

* Use `use_sigmoid` option instead of "bce" mode in label smooth loss.
 2022-09-20 15:50:21 +08:00
Ezra-Yu 624d22b441
[Refactor] Refactor Precise BN Hook. (#913) 2022-08-03 15:24:19 +08:00
mzr1996 735a3ee11f Update auto_scale_lr fields 2022-07-18 11:11:13 +08:00
yingfhu ce81a07059 [Refactor] add auto_scale_lr 2022-07-18 11:11:13 +08:00
mzr1996 c0f3ba68a8 Add MAE version ViT-base training results and fix some errors in configs. 2022-07-18 11:11:13 +08:00
mzr1996 f3299b4ca2 [Refactor] Refactor batch augmentations 2022-07-18 11:11:13 +08:00
mzr1996 548db6f4ac [Refactor] Update optimizer related registries and configs. 2022-07-18 10:53:56 +08:00
mzr1996 995b1d0d58 [Refactor] Add `ResizeEdge` and refactor all dataset configs. 2022-07-18 10:53:56 +08:00
yingfhu 65f3b2221d Modify config relates to logger and checkpoint 2022-07-18 10:53:28 +08:00
Ezra-Yu 58d9f649ed Refactor scheduler configuration 2022-07-18 10:53:28 +08:00
Ezra-Yu 2f2aa3037c Refactor default hooks configs 2022-07-18 10:53:28 +08:00
Ma Zerun 824fbcbaae
[Refactor] Use mdformat instead of markdownlint to format markdown. (#844) 
* [Refactor] Use mdformat instead of markdownlint to format markdown.

* Update unavailiable api links in tutorials

* Update CONTRIBUTING.md

* Use mdformat==0.7.9 to support Python 3.6
 2022-06-02 15:22:01 +08:00
mzr1996 f0ee5dcb2a Update checkpoint path of ResNet-50 on CUB dataset 2022-03-31 10:23:33 +08:00
takuoko aa522f4309
[Feature] Support CUB dataset. (#703) 
* support cub dataset

* support cub dataset

* fix train lint error

* add docs

* fix class label

Co-authored-by: Ezra-Yu <1105212286@qq.com>

* del debug code

* skip docformatter problem

* add unit tests

* add CUB baseline configs and chpts

* fix some typos

* fix name style

* update flops

Co-authored-by: Ezra-Yu <1105212286@qq.com>
 2022-03-16 16:22:28 +08:00
kusakana 779a06257c
[Feature] Support Wide-ResNet. (#715) 
* Add configs of Wide ResNet

* updated config

* add docs and metafile

* update configs

* remove extra import

* Update metafile and readme

Co-authored-by: Ezra-Yu <1105212286@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-03-04 14:06:34 +08:00
Ezra-Yu 7fcaedcbfb
[Feature] Add ResNetV1c. (#692) 
* add ResNetV1c

* add unit tests

* fix lint

* update docstring

* fix lint
 2022-02-23 11:36:33 +08:00
Ma Zerun 833152b1f4
[Docs] Update README in configs according to OpenMMLab standard. (#672) 
* Update README according to OpenMMLab standard.

* Update model zoo docs generation.

* Revert modification for paperlink
 2022-01-26 18:26:01 +08:00
Zhicheng Chen d56170a734
[Feature] Support EfficientNet (#649) 
* add config for resnest test

* fix config

* add label smoothing

* add memcached

* minor fix

* fix bug

* fix config

* add config

* minor fix

* fix configs

* use EResize

* change interpolation

* add more configs

* add docsting

* add unittest

* remove unnecessary changes

* minor fix

* add more docstring

* fix linting

* add efficient backbone

* add config

* add Edge Residual

* fix bug

* remove unnecessary files

* refactor

* add resize in crop to ensure crop size is output size

* fix bug and add comments

* test

* fix

* add more configs

* add more configs

* add more configs

* fix bug

* add model zoo

* fix

* reorganize code

* add edge tpu

* add edge tpu converter

* rename

* update readme

* reorganize code and config

* Rename configs of EfficientNet, and add metafile & model_zoo

* Remove `backend='pillow'`

* Add comments about EfficientNet-EdgeTPU

* Rename the convert tool of EfficientNet.

* Refactor EfficientNet and update docstring.

* Update EfficientNet-EdgeTPU config

* Fix unit tests

Co-authored-by: lixinran <lixr423@outlook.com>
Co-authored-by: lixinran <lixinran@sensetime.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-01-25 12:14:17 +08:00
mzr1996 cdf569a805 Fix ResNet rsb config 2021-12-31 11:12:21 +08:00
mzr1996 e1a4020168 Fix ResNet rsb config 2021-12-31 11:08:11 +08:00
Zhicheng Chen 0bbbb04429
[Enhancement] Improve the accuracy of ResNet (#572) 
* add itp timm

* minor update

* minor update

* minor update

* add rep aug, minor update on configs

* minor update

* add target threshold

* add decaymulti

* minor update

* minor update

* add lbl smooth

* update lr

* reorganize config files and code

* minor bugfixes

* remove unused parts and minor fixes on cfg

* critical bugfix, add test and cfg update

* refactor code

* update doc string

* remove duplicate code

* refactor drop path in resnet

* rename

* Modify configs and add README&metafile

* Update metafile

Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-12-29 10:58:42 +08:00
Ma Zerun 159b38d276
[Reproduction] Reproduce training results of T2T-ViT (#610) 
* Add cosine cool down lr updater

* Use ema hook

* Update decay mult

* Update configs.

* Update T2T-ViT readme and format all readme

* Update swin readme

* Update tnt readme

* Add docstring for `CosineAnnealingCooldownLrUpdaterHook`.

* Update t2t readme and metafile
 2021-12-28 15:09:40 +08:00
Ezra-Yu d1d9ebfc67
[Feature] Add Precise BN (#401) 
* add preciseBN hook

* add precise-bn config

* add proority of preciseBN

* Add precise BN hook

* Update PreciseBN hook

* add config example & pre-commit

* Update PBN

* Add unit test

* Update unit tests

* mv preciseBN to mmcls.runner

* rm extra sleep()

* merge latest

* update code

* fix lint

* update unit tests

* update unit tests and add warning

* update docstring and renamme filename

* remove model warpper

* add copyright

* Imporve docs and update some variable names

* Update resnet50_8xb32-coslr-preciseBN_in1k.py

Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-12-24 16:28:42 +08:00
Ma Zerun f3fbc8b90b
[Docs] Add abstract and image for every paper. (#546) 2021-11-24 17:23:37 +08:00
Ma Zerun 7977dc8e2d
[Improvement] Rename config files according to the config name standard. (#508) 
* Update tnt config

* Rename config files

* Update docs

* Update metafile in dev_scripts

* Fix some files

* Remove fp16 metafile and README.

* Fix names in metafiles.
 2021-11-19 14:20:35 +08:00
huoshuai-dot 3f52488d52
[Docs] Add the top-5 acc in ResNet-CIFAR README. (#531) 
* Update README.md

* Update README.md
 2021-11-12 16:13:41 +08:00
Ezra-Yu 2ce5825ef1
[Feature] Support ImageNet21k dataset. (#461) 
* add imagnet21k

* Update unit test

* update imaenet21k

* use slots

* use slots

* rename Data_item to ImageInfo

* add unit tests

* Update unit tests

* rm some print

* update unit tests

* fix lint

* remove default value of pipeline
 2021-10-28 15:22:08 +08:00
Ma Zerun fd0f5cce92
[Docs] Add model-pages in Model Zoo (#480) 
* Add model-pages

* Add shortname in configs

* Use link directly instead of `switch_language.md`

* Auto collapse model-zoo pages.

* Fix link in RegVGG

* Add link replace

* fix lint
 2021-10-14 15:26:47 +08:00
Ma Zerun 8f68779cc6
[Docs] Update metafile and Readme (#435) 
* Update metafile format.

* Update accuracy of checkpoints.

* Add metafile and readme for tnt.

* Add converted ckpts in swin-transformer.

* Fix tnt ckpt link

* Update swin_transformer metafile
 2021-09-18 16:32:46 +08:00
Ma Zerun 57e9692765
Fix broken _base_ link in a resnet config. (#361) 2021-07-25 23:50:32 -04:00
whcao 5a86288f1f
[Fix]Fix metafile (#335) 
* add mytrain.py for test

* test before layers

* test attr in layers

* test classifier

* delete mytrain.py

* add mobilenet_v3.yml

* fix ymal

* rename model_zoo.yml to model-index.yml

* replace imagenet with imagenet-1k
 2021-07-01 20:50:42 +08:00
LXXXXR c68bf9f2ba
fix resnetv1d configs and update ckpts (#288) 2021-06-03 16:30:29 +08:00
LXXXXR 5ff158cec9
[Bug] Fix Cifar color channel (#280) 
* fix bug

* updatedoc accordingly
 2021-05-31 14:15:22 +08:00
LXXXXR 5fb9ced5f9
[Docs] Change tag into comment (#226) 
* change tags

* remove modelzoo_statistics
 2021-04-28 17:08:57 +08:00
LXXXXR a604cd97ef
[Feature] Adding PWC metafile (#225) 
* add yml

* fix VGG

* minor fix

* add model zoo

* add bs and epoch

* revised according to comments

* revised according to comments

* revised according to comments
 2021-04-28 17:07:26 +08:00
whcao 1cde6f6e65
[Feature] Add cutmix option (#198) 
* Add cutmix option

* fix code style

* add some annotations

* add annotation about custom_hooks

* check constraint of alpha > 0

* add test cutmix

* fix code style

* add cutmix to configs/models

* add cutmix to configs/resnet

* flake8

* empty
 2021-04-14 21:27:42 +08:00
LXXXXR 3f085026cf
[Feature] Add Cifar100 config (#208) 
* add config for cifar100

* fix doc
 2021-04-13 20:15:29 +08:00
whcao af83e981ac
[Bug]Fix label smooth bug (#203) 
* add convert_to_one_hot

* add test_label_smooth_loss

* add my label_smooth_loss

* fix CELoss bug

* test new label smooth loss

* LabelSmoothLoss downward compatibility

* add some comments

* remove the old version of LabelSmoothLoss

* add some comments

* add some comments

* add some comments

* add label smooth to config
 2021-04-13 13:53:56 +08:00
ftbabi bdd6b01ae7
[Feature] Add "mixup" from Bag of Tricks (#160) 
* Add mixup option

* Modify the structure of mixup and add configs

* Clean configs

* Add test for mixup and SoftCrossEntropyLoss

* Add simple test for ImageClassifier

* Fix bug in test_losses.py

* Add assertion in CrossEntropyLoss
 2021-02-25 14:06:58 +08:00
Cheng-Yen Yang 87d4a1fcea
Update README.md (#150) 2021-01-25 17:16:57 +08:00
LXXXXR aa50103442
fix typo in configs (#147) 2021-01-21 14:47:02 +08:00
congee c8109d1eca
Update stat to classify papers (#139) 
* update stat to classify papers

* modify tag of papers
 2021-01-14 16:48:05 +08:00
LXXXXR d062cdef0a
[Docs] Add tags in README.md (#137) 
* add tag in README.md

* fix linting issue
 2021-01-11 11:10:06 +08:00
LXXXXR 736eec1fd2
Fix links in docs (#118) 
* fix links in docs

* Revised according to comments

* revised according to comments
 2020-12-21 15:29:12 +08:00
LXXXXR 668292f5ab
Add README.md in configs (#117) 
* rearrange config files

* rearrange config files

* add README.md in configs

* rename
 2020-12-18 20:04:54 +08:00