Skip to content

Commit 9fc33f3

Browse files
committed
update README.md
1 parent 461fffb commit 9fc33f3

File tree

1 file changed

+42
-2
lines changed

1 file changed

+42
-2
lines changed

README.md

Lines changed: 42 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,5 +27,45 @@ The core concept of DyRep is in `lib/models/utils/dyrep.py`.
2727
```
2828
* CIFAR-100
2929
```
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.

0 commit comments

Comments
 (0)