mmpretrain

Commit Graph

Author	SHA1	Message	Date
Ma Zerun	9fd35dd7b5	[Reproduction] Reproduce training results of DeiT. (#711 ) * Update deit training settings * Update decay config * Add mixup&cutmix and drop path rate * Update training configs * Update model-zoo * Add comments	2022-03-02 14:23:10 +08:00
Ma Zerun	833152b1f4	[Docs] Update README in configs according to OpenMMLab standard. (#672 ) * Update README according to OpenMMLab standard. * Update model zoo docs generation. * Revert modification for paperlink	2022-01-26 18:26:01 +08:00
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
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

5 Commits (9fd35dd7b5da7a298c70f1947a8991462a896910)