mmpretrain

Commit Graph

Author	SHA1	Message	Date
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

Author

SHA1

Message

Date

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

1 Commits (f9a2b04cee1cd6a416b6b2731e7112c12520dda5)