mmpretrain/tests
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
..
data [Feature] Add a tool to visualize learning rate in each iterations (#498) 2021-11-03 10:48:56 +08:00
test_data [Fix] Fix bug when using `ClassBalancedDataset` (#555) 2021-11-26 09:40:00 +08:00
test_downstream [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_metrics [Enhance] Imporve efficiency of precision, recall, f1_score and support. (#595) 2021-12-13 17:24:59 +08:00
test_models [Feature] Add DeiT backbone and checkpoints. (#576) 2021-12-15 22:44:57 +08:00
test_runtime [Feature] Support LAMB optimizer. (#591) 2021-12-13 17:24:44 +08:00
test_utils [Refactor] Change `load_json_logs` to `load_json_log`. (#602) 2021-12-14 18:03:27 +08:00