File tree Expand file tree Collapse file tree 1 file changed +42
-2
lines changed Expand file tree Collapse file tree 1 file changed +42
-2
lines changed Original file line number Diff line number Diff line change @@ -27,5 +27,45 @@ The core concept of DyRep is in `lib/models/utils/dyrep.py`.
27
27
```
28
28
* CIFAR-100
29
29
```
30
- sh tools/dist_train.sh 1 configs/strategies/DyRep/cifar.yaml nas_model --model-config configs/models/VGG/vgg16_cifar10.yaml --dyrep --experiment dyrep_cifar10_vgg16
31
- ```
30
+ sh tools/dist_train.sh 1 configs/strategies/DyRep/cifar.yaml nas_model --model-config configs/models/VGG/vgg16_cifar100.yaml --dyrep --dataset cifar100 --experiment dyrep_cifar100_vgg16
31
+ ```
32
+
33
+ ### ImageNet
34
+ * ResNets
35
+ ```
36
+ sh tools/dist_train.sh 8 configs/strategies/DyRep/resnet.yaml resnet50 --dyrep --experiment dyrep_imagenet_res50
37
+ ```
38
+
39
+ * MobileNetV1
40
+ ```
41
+ sh tools/dist_train.sh 8 configs/strategies/DyRep/mbv1.yaml mobilenet_v1 --dyrep --experiment dyrep_imagenet_mbv1
42
+ ```
43
+
44
+ * RepVGG
45
+ * DyRep-A2
46
+ ```
47
+ sh tools/dist_train.sh 8 configs/strategies/DyRep/repvgg_baseline.yaml timm_repvgg_a2 --dyrep --dyrep_recal_bn_every_epoch --experiment dyrep_imagenet_repvgg_a2
48
+ ```
49
+ * DyRep-B2g4 and DyRep-B3
50
+ ```
51
+ sh tools/dist_train.sh 8 configs/strategies/DyRep/repvgg_strong.yaml timm_repvgg_b2g4 --dyrep --dyrep_recal_bn_every_epoch --experiment dyrep_imagenet_repvgg_b2g4
52
+ ```
53
+
54
+ ## Deploying the Trained DyRep Models to Inference Models
55
+ ```
56
+ sh tools/dist_convert.sh 8 ${CONFIG} ${MODEL} --resume ${CHECKPOINT}
57
+ ```
58
+
59
+ For example, if you want to deploy the trained ResNet-50 model with the best checkpoint, run
60
+ ```
61
+ sh tools/dist_convert.sh 8 configs/strategies/DyRep/resnet.yaml resnet50 --dyrep --resume experiments/dyrep_imagenet_res50/best.pth.tar
62
+ ```
63
+
64
+ Then it will run test before and after deployment to ensure the accuracy will not drop.
65
+
66
+ The final weights of the inference model will be saved in `experiments/dyrep_imagenet_res50/convert/model.ckpt`.
67
+
68
+ ## Results
69
+
70
+ ## Citation
71
+ The paper will be released soon.
You can’t perform that action at this time.
0 commit comments