16947f1239
* 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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| alexnet.py | ||
| base_backbone.py | ||
| lenet.py | ||
| mobilenet_v2.py | ||
| mobilenet_v3.py | ||
| regnet.py | ||
| resnest.py | ||
| resnet.py | ||
| resnet_cifar.py | ||
| resnext.py | ||
| seresnet.py | ||
| seresnext.py | ||
| shufflenet_v1.py | ||
| shufflenet_v2.py | ||
| vgg.py | ||
| vision_transformer.py | ||