5 Commits

Author SHA1 Message Date
mzr1996
995b1d0d58 [Refactor] Add ResizeEdge and refactor all dataset configs. 2022-07-18 10:53:56 +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
f9a2b04cee
[Feature] Add DeiT backbone and checkpoints. (#576)
* Support DeiT backbone.

* Use hook to automatically resize pos embed

* Update ViT training setting

* Add deit configs and update docs

* Fix vit arch assertion

* Remove useless init function

* Add unit tests.

* Fix resize_pos_embed for DeiT

* Improve according to comments.
2021-12-15 22:44:57 +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