Ma Zerun 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

..

__init__.py

[Reproduction] Reproduce training results of T2T-ViT (#610)

2021-12-28 15:09:40 +08:00

class_num_check_hook.py

[Enhance] Added NumClassCheckHook and unit tests (#559)

2021-12-08 18:15:05 +08:00

lr_updater.py

[Reproduction] Reproduce training results of T2T-ViT (#610)

2021-12-28 15:09:40 +08:00

precise_bn_hook.py

[Feature] Add Precise BN (#401)

2021-12-24 16:28:42 +08:00