"ts/webui/src/vscode:/vscode.git/clone" did not exist on "ff1af7f2627e07b229265f8e6ad8e536bf5d2827"
model_compression.sh 1.33 KB
Newer Older
chicm-ms's avatar
chicm-ms committed
1
2
3
4
5
6
7
8
#!/bin/bash
set -e
CWD=${PWD}

echo ""
echo "===========================Testing: pruning and speedup==========================="
cd ${CWD}/../examples/model_compress

chicm-ms's avatar
chicm-ms committed
9
10
11
12
13
14
15
16
for name in fpgm slim l1filter apoz
do
    echo "testing $name pruning and speedup..."
    python3 model_prune_torch.py --pruner_name $name --pretrain_epochs 1 --prune_epochs 1
    python3 model_speedup.py --example_name $name
done

for name in level mean_activation
chicm-ms's avatar
chicm-ms committed
17
18
19
20
21
do
    echo "testing $name pruning..."
    python3 model_prune_torch.py --pruner_name $name --pretrain_epochs 1 --prune_epochs 1
done

chicm-ms's avatar
chicm-ms committed
22
23
24
25
26
27
28
29
30
echo 'testing level pruner pruning'
python3 model_prune_torch.py --pruner_name level --pretrain_epochs 1 --prune_epochs 1

echo 'testing agp pruning'
python3 model_prune_torch.py --pruner_name agp --pretrain_epochs 1 --prune_epochs 2

echo 'testing mean_activation pruning'
python3 model_prune_torch.py --pruner_name mean_activation --pretrain_epochs 1 --prune_epochs 1

chicm-ms's avatar
chicm-ms committed
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
#echo "testing lottery ticket pruning..."
#python3 lottery_torch_mnist_fc.py

echo ""
echo "===========================Testing: quantizers==========================="
# to be enabled
#echo "testing QAT quantizer..."
#python3 QAT_torch_quantizer.py

#echo "testing DoReFa quantizer..."
#python3 DoReFaQuantizer_torch_mnist.py

#echo "testing BNN quantizer..."
#python3 BNN_quantizer_cifar10.py

rm -rf ./checkpoints/*