9 Commits

Author SHA1 Message Date
mzr1996
522ab1fd84 [Feature] Add Visualization hook 2022-07-18 11:11:04 +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
Ezra-Yu
4d4b22377d Add browse dataset 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
2f2aa3037c Refactor default hooks configs 2022-07-18 10:53:28 +08:00
whcao
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
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
yanglei
5c88675a71 Add configs for MNIST, CIFAR10 and ImageNet 2020-07-12 00:06:57 +08:00