Merge branch 'master' of https://github.com/sundw2014/DKS

sundw2014 · Jun 4, 2019 · 350f2b6 · 350f2b6
2 parents dfc25a9 + d595293
commit 350f2b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -76,6 +76,6 @@ python main.py --save ./logs/wrn-28-10_dropout --depth 28 --width 10 --dropout -
 
  ```
  cd imagenet/
- python main.py -a [resnet18/50/152] --mode_path [path to the pretrained model] [ImageNet folder with the val subfolder]
+ python main.py -a [resnet18/50/152] --model_path [path to the pretrained model] [ImageNet folder with the val subfolder]
  ```