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