Change checkpoint path in pruning demo. (#3888)

532ebad1 · whs · GitHub · c5917836 · 532ebad1
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

PaddleSlim/classification/pruning/configs/mobilenet_v2.yaml PaddleSlim/classification/pruning/configs/mobilenet_v2.yaml +1 -1

未找到文件。
--- a/PaddleSlim/classification/pruning/configs/mobilenet_v2.yaml
+++ b/PaddleSlim/classification/pruning/configs/mobilenet_v2.yaml
@@ -17,6 +17,6 @@ strategies:
 #        pruned_params: '.*expand_weights'
 compressor:
    epoch: 241
-    checkpoint_path: './checkpoints/'
+    checkpoint_path: './checkpoints/mobilenet_v2'
    strategies:
        - uniform_pruning_strategy