From a503f403c15b9c1cdc618aa066dc970e221b563d Mon Sep 17 00:00:00 2001 From: Guanhua Wang Date: Wed, 9 Sep 2020 17:33:10 -0700 Subject: [PATCH] Update README.md --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 5748a2e..e72553b 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,7 @@ Supported CNN architectures and datasets: python3 group_selection.py --arch $ARCH --dataset imagenet --ngroups $number_of_groups --gpu_num $number_of_gpu --data /{path_to_imagenet_dataset} ``` - Pruning candidate now stored in `./prune_candidate_logs` + Pruning candidate now stored in `./prune_candidate_logs/` ### 2. Prune models: @@ -49,7 +49,7 @@ Supported CNN architectures and datasets: python3 prune_and_get_model.py -a $ARCH --dataset imagenet -c ./prune_candidate_logs/ -s ./{TO_SAVE_PRUNED_MODEL_DIR} --pretrained ``` - Pruned models are now saved in ./TO_SAVE_PRUNED_MODEL_DIR/$ARCH + Pruned models are now saved in `./TO_SAVE_PRUNED_MODEL_DIR/$ARCH/` ### 3. Retrain pruned models: @@ -62,7 +62,7 @@ Supported CNN architectures and datasets: python3 retrain_grouped_model.py -a $ARCH --dataset imagenet --resume ./{TO_SAVE_PRUNED_MODEL_DIR}/ --epochs $number_of_epochs --num_gpus $number_of_gpus --train_batch $batch_size --data /{path_to_imagenet_dataset} ``` - Retrained models now saved in ./TO_SAVE_PRUNED_MODEL_DIR_retrained/$ARCH/ + Retrained models now saved in `./{TO_SAVE_PRUNED_MODEL_DIR}_retrained/$ARCH/` ### 4. Evaluate