735a3ee11f
Update auto_scale_lr fields
2022-07-18 11:11:13 +08:00
ce81a07059
[Refactor] add auto_scale_lr
2022-07-18 11:11:13 +08:00
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
548db6f4ac
[Refactor] Update optimizer related registries and configs.
2022-07-18 10:53:56 +08:00
995b1d0d58
[Refactor] Add `ResizeEdge` and refactor all dataset configs.
2022-07-18 10:53:56 +08:00
58d9f649ed
Refactor scheduler configuration
2022-07-18 10:53:28 +08:00
159b38d276
[Reproduction] Reproduce training results of T2T-ViT ( #610 )
...
* Add cosine cool down lr updater
* Use ema hook
* Update decay mult
* Update configs.
* Update T2T-ViT readme and format all readme
* Update swin readme
* Update tnt readme
* Add docstring for `CosineAnnealingCooldownLrUpdaterHook`.
* Update t2t readme and metafile
2021-12-28 15:09:40 +08:00
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
mzr1996
yingfhu
Ezra-Yu