mmpretrain

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	643fb192cd	[Enhance] Enhance feature extraction function. (#593 ) * Fix MobileNet V3 configs * Refactor to support more powerful feature extraction. * Add unit tests * Fix unit test * Imporve according to comments * Update checkpoints path * Fix unit tests * Add docstring of `simple_test` * Add docstring of `extract_feat` * Update model zoo	2021-12-17 15:55:02 +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

Author

SHA1

Message

Date

Ma Zerun

643fb192cd

[Enhance] Enhance feature extraction function. (#593 )

* Fix MobileNet V3 configs

* Refactor to support more powerful feature extraction.

* Add unit tests

* Fix unit test

* Imporve according to comments

* Update checkpoints path

* Fix unit tests

* Add docstring of `simple_test`

* Add docstring of `extract_feat`

* Update model zoo

2021-12-17 15:55:02 +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

2 Commits (a19c28fe9538fd1c06dbc08c4dc7ad59d67d4217)