diff --git a/PaddleSlim/configs/filter_pruning_sen.yaml b/PaddleSlim/configs/filter_pruning_sen.yaml index 4537405fe200fad631095d0bd81d3de2fcb267bc..f83079f5ff704f0e8605bb41e010e935a4180a44 100644 --- a/PaddleSlim/configs/filter_pruning_sen.yaml +++ b/PaddleSlim/configs/filter_pruning_sen.yaml @@ -20,6 +20,7 @@ strategies: metric_name: 'acc_top1' compressor: epoch: 200 + #init_model: './checkpoints/0' # Please enable this option for loading checkpoint. checkpoint_path: './checkpoints/' strategies: - sensitive_pruning_strategy diff --git a/PaddleSlim/configs/filter_pruning_uniform.yaml b/PaddleSlim/configs/filter_pruning_uniform.yaml index 1dea1070f4456e7150ed1a2706cde072180bf9ff..798e7f1bea875d626e84fd802df885ed95194de6 100644 --- a/PaddleSlim/configs/filter_pruning_uniform.yaml +++ b/PaddleSlim/configs/filter_pruning_uniform.yaml @@ -16,6 +16,7 @@ strategies: metric_name: 'acc_top1' compressor: epoch: 200 + #init_model: './checkpoints/0' # Please enable this option for loading checkpoint. checkpoint_path: './checkpoints/' strategies: - uniform_pruning_strategy diff --git a/PaddleSlim/configs/quantization_pruning.yaml b/PaddleSlim/configs/quantization_pruning.yaml index b4dcd4794dc244c00c144d2fd02bbd7c859c461e..0499051037215c8341ef18fdffb8605b912ee8c4 100644 --- a/PaddleSlim/configs/quantization_pruning.yaml +++ b/PaddleSlim/configs/quantization_pruning.yaml @@ -30,6 +30,7 @@ strategies: activation_quantize_type: 'abs_max' compressor: epoch: 142 + #init_model: './checkpoints/0' # Please enable this option for loading checkpoint. checkpoint_path: './checkpoints/' strategies: - uniform_pruning_strategy diff --git a/PaddleSlim/run.sh b/PaddleSlim/run.sh index eaed30b84cc70805db1738f949bae674f207f17d..0d370e98de4a538f21d1c5d8be43d43cb000252f 100644 --- a/PaddleSlim/run.sh +++ b/PaddleSlim/run.sh @@ -24,6 +24,10 @@ fi cd - +# enable GC strategy +export FLAGS_fast_eager_deletion_mode=1 +export FLAGS_eager_delete_tensor_gb=0.0 + # for distillation #----------------- export CUDA_VISIBLE_DEVICES=0,1,2,3