Skip to content

GitLab

Unverified Commit 727ee1ce authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Fix IT on model compression (#3673)

parent 4c49db12
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 13 deletions
test/scripts/model_compression.sh
View file @ 727ee1ce
...@@ -4,35 +4,35 @@ CWD=${PWD} ...@@ -4,35 +4,35 @@ CWD=${PWD}
echo "" echo ""
echo "===========================Testing: pruning and speedup===========================" echo "===========================Testing: pruning and speedup==========================="
cd ${CWD}/../examples/model_compress cd ${CWD}/../examples/model_compress/pruning
echo "testing fpgm pruning and speedup..." echo "testing fpgm pruning and speedup..."
python3 pruning/basic_pruners_torch.py --pruner fpgm --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10 python3 basic_pruners_torch.py --pruner fpgm --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10
python3 pruning/model_speedup.py --example_name fpgm python3 model_speedup.py --example_name fpgm
echo "testing slim pruning and speedup..." echo "testing slim pruning and speedup..."
python3 pruning/basic_pruners_torch.py --pruner slim --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg19 --dataset cifar10 --sparsity 0.7 python3 basic_pruners_torch.py --pruner slim --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg19 --dataset cifar10 --sparsity 0.7
python3 pruning/model_speedup.py --example_name slim python3 model_speedup.py --example_name slim
echo "testing l1filter pruning and speedup..." echo "testing l1filter pruning and speedup..."
python3 pruning/basic_pruners_torch.py --pruner l1filter --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10 python3 basic_pruners_torch.py --pruner l1filter --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10
python3 pruning/model_speedup.py --example_name l1filter python3 model_speedup.py --example_name l1filter
echo "testing apoz pruning and speedup..." echo "testing apoz pruning and speedup..."
python3 pruning/basic_pruners_torch.py --pruner apoz --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10 python3 basic_pruners_torch.py --pruner apoz --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10
python3 pruning/model_speedup.py --example_name apoz python3 model_speedup.py --example_name apoz
echo 'testing level pruner pruning' echo 'testing level pruner pruning'
python3 pruning/basic_pruners_torch.py --pruner level --pretrain-epochs 1 --fine-tune-epochs 1 --model lenet --dataset mnist python3 basic_pruners_torch.py --pruner level --pretrain-epochs 1 --fine-tune-epochs 1 --model lenet --dataset mnist
echo 'testing agp pruning' echo 'testing agp pruning'
python3 pruning/basic_pruners_torch.py --pruner agp --pretrain-epochs 1 --fine-tune-epochs 1 --model lenet --dataset mnist python3 basic_pruners_torch.py --pruner agp --pretrain-epochs 1 --fine-tune-epochs 1 --model lenet --dataset mnist
echo 'testing mean_activation pruning' echo 'testing mean_activation pruning'
python3 pruning/basic_pruners_torch.py --pruner mean_activation --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10 python3 basic_pruners_torch.py --pruner mean_activation --pretrain-epochs 1 --fine-tune-epochs 1 --model vgg16 --dataset cifar10
echo "testing lottery ticket pruning..." echo "testing lottery ticket pruning..."
python3 pruning/lottery_torch_mnist_fc.py --train_epochs 1 python3 lottery_torch_mnist_fc.py --train_epochs 1
echo "" echo ""
echo "===========================Testing: quantizers===========================" echo "===========================Testing: quantizers==========================="
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment