mmclassification/mmcls/models
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
..
backbones [Feature] Add DeiT backbone and checkpoints. (#576) 2021-12-15 22:44:57 +08:00
classifiers [Fix] Fix a bug when using iter-based runner with 'val' workflow (#542) 2021-11-22 12:06:47 +08:00
heads [Feature] Add DeiT backbone and checkpoints. (#576) 2021-12-15 22:44:57 +08:00
losses [Fix] Fix cal_acc bug caused by the return type of accuracy. 2021-12-14 16:21:25 +08:00
necks [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
utils [Refactor] Refator ViT (Continue #295) (#395) 2021-10-18 16:07:00 +08:00
__init__.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
builder.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00