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
0979e78573
Rename the package name to `mmpretrain`.
2023-02-17 15:20:55 +08:00
f9be21ab74
[Docs] Add version selection in the banner. ( #1217 )
...
* [Docs] Add version selection in the banner.
* Small fix.
2022-11-23 13:38:23 +08:00
c0f3ba68a8
Add MAE version ViT-base training results and fix some errors in configs.
2022-07-18 11:11:13 +08:00
522ab1fd84
[Feature] Add Visualization hook
2022-07-18 11:11:04 +08:00
548db6f4ac
[Refactor] Update optimizer related registries and configs.
2022-07-18 10:53:56 +08:00
995b1d0d58
[Refactor] Add `ResizeEdge` and refactor all dataset configs.
2022-07-18 10:53:56 +08:00
4d4b22377d
Add browse dataset
2022-07-18 10:53:56 +08:00
65f3b2221d
Modify config relates to logger and checkpoint
2022-07-18 10:53:28 +08:00
2f2aa3037c
Refactor default hooks configs
2022-07-18 10:53:28 +08:00
3be95b99c2
[Feature]Modify modelzoo readme ( #230 )
...
* add mytrain.py for test
* test before layers
* test attr in layers
* test classifier
* delete mytrain.py
* add imagenet_bs4096_AdamW.py
* delete 2 lines of comments
* change bs to 64
* fix bug
* add vit to model_zoo.md
* rename
2021-04-29 15:18:55 +08:00
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
5c88675a71
Add configs for MNIST, CIFAR10 and ImageNet
2020-07-12 00:06:57 +08:00
Ma Zerun
Ezra-Yu
yingfhu
whcao
yanglei