Ma Zerun
|
fffa30dd48
|
[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467)
* add t2t backbone
* register t2t_vit
* add t2t_vit config
* [Temp] Align posterize transform with timm.
* Fix lint
* Refactor t2t-vit
* Add config for t2t-vit
* Add metafile and README for t2t-vit
* Add unit tests
* configs
* Update metafile and README
* Improve docstring
* Fix batch size which should be 8x64 instead of 8x128
* Fix typo
* Update model zoo
* Update training augments config.
* Move some arguments of T2TModule to T2TViT
* Update docs.
* Update unit test
Co-authored-by: HIT-cwh <2892770585@qq.com>
|
2021-10-29 10:37:16 +08:00 |