mmpretrain

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	c708770b42	[Enhance] Support dynamic input shape for ViT-based algorithms. (#706 ) * Move `resize_pos_embed` to `mmcls.models.utils` * Refactor Vision Transformer * Refactor DeiT * Refactor MLP-Mixer * Refactor Swin-Transformer * Remove `indexing` arg * Support dynamic inputs for t2t_vit * Add copyright * Fix bugs in swin transformer * Add `pad_small_maps` option * Update swin transformer * Handle `attn_mask` in checkpoints of swin * Imporve by comments	2022-03-03 13:10:12 +08:00
xcnick	3eafc5cbe1	[Feature] Add MLP Mixer Backbone. (#528 ) * Add Mlp Mixer * Fix token_mix FNN add_identity and add out_indices * Fix mlp_mixer unit test	2021-11-15 18:25:22 +08:00

Author

SHA1

Message

Date

Ma Zerun

c708770b42

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706 )

* Move `resize_pos_embed` to `mmcls.models.utils`

* Refactor Vision Transformer

* Refactor DeiT

* Refactor MLP-Mixer

* Refactor Swin-Transformer

* Remove `indexing` arg

* Support dynamic inputs for t2t_vit

* Add copyright

* Fix bugs in swin transformer

* Add `pad_small_maps` option

* Update swin transformer

* Handle `attn_mask` in checkpoints of swin

* Imporve by comments

2022-03-03 13:10:12 +08:00

xcnick

3eafc5cbe1

[Feature] Add MLP Mixer Backbone. (#528 )

* Add Mlp Mixer

* Fix token_mix FNN add_identity and add out_indices

* Fix mlp_mixer unit test

2021-11-15 18:25:22 +08:00

2 Commits (556fa567a8c7807a8dfde13edb01eea49ec16d03)