mmpretrain

Commit Graph

Author	SHA1	Message	Date
mzr1996	735a3ee11f	Update auto_scale_lr fields	2022-07-18 11:11:13 +08:00
mzr1996	24bcf069f8	[Refactor] Refactor dev scripts	2022-07-18 11:11:13 +08:00
mzr1996	c0f3ba68a8	Add MAE version ViT-base training results and fix some errors in configs.	2022-07-18 11:11:13 +08:00
mzr1996	548db6f4ac	[Refactor] Update optimizer related registries and configs.	2022-07-18 10:53:56 +08:00
Ezra-Yu	58d9f649ed	Refactor scheduler configuration	2022-07-18 10:53:28 +08:00
Ezra-Yu	2f2aa3037c	Refactor default hooks configs	2022-07-18 10:53:28 +08:00
whcao	16947f1239	[Bug]Fix weight decay (#227 ) * add imagenet bs 4096 * add vit_base_patch16_224_finetune * add vit_base_patch16_224_pretrain * add vit_base_patch16_384_finetune * add vit_base_patch16_384_finetune * add vit_b_p16_224_finetune_imagenet * add vit_b_p16_224_pretrain_imagenet * add vit_b_p16_384_finetune_imagenet * add vit * add vit * add vit head * vit unitest * keep up with ClsHead * test vit * add flag to determiine whether to calculate acc during training * Changes related to mmcv1.3.0 * change checkpoint saving interval to 10 * add label smooth * default_runtime.py recovery * docformatter * docformatter * delete 2 lines of comments * delete configs/_base_/schedules/imagenet_bs4096.py * add configs/_base_/schedules/imagenet_bs2048_AdamW.py * rename imagenet_bs4096.py to imagenet_bs2048_AdamW.py * add AutoAugment * fix weight decay in vit * change eval interval to 10 * add mytrain.py for test * test before layers * test attr in layers * test classifier * delete mytrain.py * delete @torch.jit.ignore * change eval interval back to 1 * add some comments to imagenet_bs2048_AdamW * add some comments	2021-04-28 17:16:43 +08:00
whcao	affb39fe07	[Feature]Add Vit (#214 ) * add imagenet bs 4096 * add vit_base_patch16_224_finetune * add vit_base_patch16_224_pretrain * add vit_base_patch16_384_finetune * add vit_base_patch16_384_finetune * add vit_b_p16_224_finetune_imagenet * add vit_b_p16_224_pretrain_imagenet * add vit_b_p16_384_finetune_imagenet * add vit * add vit * add vit head * vit unitest * keep up with ClsHead * test vit * add flag to determiine whether to calculate acc during training * Changes related to mmcv1.3.0 * change checkpoint saving interval to 10 * add label smooth * default_runtime.py recovery * docformatter * docformatter * delete 2 lines of comments * delete configs/_base_/schedules/imagenet_bs4096.py * add configs/_base_/schedules/imagenet_bs2048_AdamW.py * rename imagenet_bs4096.py to imagenet_bs2048_AdamW.py * add helpers.py * test vit hybrid backbone * fix HybridEmbed * use to_2tuple instead	2021-04-16 19:22:41 +08:00

8 Commits (735a3ee11f857bb50039d946538917ccbace448d)