Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
nni
Commits
30df69bf
"...git@developer.sourcefind.cn:modelzoo/qwen_lmdeploy.git" did not exist on "74a4f3c91b5083009a55bfc4be4a3f555034fb4a"
Unverified
Commit
30df69bf
authored
Jul 16, 2021
by
Ningxin Zheng
Committed by
GitHub
Jul 16, 2021
Browse files
Fix integration test on speedup examples (#3945)
parent
a7278d2d
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
8 additions
and
8 deletions
+8
-8
docs/en_US/Compression/ModelSpeedup.rst
docs/en_US/Compression/ModelSpeedup.rst
+2
-2
examples/model_compress/pruning/speedup/model_speedup.py
examples/model_compress/pruning/speedup/model_speedup.py
+2
-2
test/scripts/model_compression.sh
test/scripts/model_compression.sh
+4
-4
No files found.
docs/en_US/Compression/ModelSpeedup.rst
View file @
30df69bf
...
@@ -37,7 +37,7 @@ Usage
...
@@ -37,7 +37,7 @@ Usage
out = model(dummy_input)
out = model(dummy_input)
print('elapsed time: ', time.time() - start)
print('elapsed time: ', time.time() - start)
For complete examples please refer to :githublink:`the code <examples/model_compress/pruning/model_speedup.py>`
For complete examples please refer to :githublink:`the code <examples/model_compress/pruning/
speedup/
model_speedup.py>`
NOTE: The current implementation supports PyTorch 1.3.1 or newer.
NOTE: The current implementation supports PyTorch 1.3.1 or newer.
...
@@ -51,7 +51,7 @@ For PyTorch we can only replace modules, if functions in ``forward`` should be r
...
@@ -51,7 +51,7 @@ For PyTorch we can only replace modules, if functions in ``forward`` should be r
Speedup Results of Examples
Speedup Results of Examples
---------------------------
---------------------------
The code of these experiments can be found :githublink:`here <examples/model_compress/pruning/model_speedup.py>`.
The code of these experiments can be found :githublink:`here <examples/model_compress/pruning/
speedup/
model_speedup.py>`.
slim pruner example
slim pruner example
^^^^^^^^^^^^^^^^^^^
^^^^^^^^^^^^^^^^^^^
...
...
examples/model_compress/pruning/speedup/model_speedup.py
View file @
30df69bf
...
@@ -7,7 +7,7 @@ import torch.nn.functional as F
...
@@ -7,7 +7,7 @@ import torch.nn.functional as F
from
torchvision
import
datasets
,
transforms
from
torchvision
import
datasets
,
transforms
import
sys
import
sys
sys
.
path
.
append
(
'../
../
models'
)
sys
.
path
.
append
(
'../models'
)
from
cifar10.vgg
import
VGG
from
cifar10.vgg
import
VGG
from
mnist.lenet
import
LeNet
from
mnist.lenet
import
LeNet
...
...
test/scripts/model_compression.sh
View file @
30df69bf
...
@@ -8,19 +8,19 @@ cd ${CWD}/../examples/model_compress/pruning
...
@@ -8,19 +8,19 @@ cd ${CWD}/../examples/model_compress/pruning
echo
"testing fpgm pruning and speedup..."
echo
"testing fpgm pruning and speedup..."
python3 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 model_speedup.py
--example_name
fpgm
python3
speedup/
model_speedup.py
--example_name
fpgm
echo
"testing slim pruning and speedup..."
echo
"testing slim pruning and speedup..."
python3 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 model_speedup.py
--example_name
slim
python3
speedup/
model_speedup.py
--example_name
slim
echo
"testing l1filter pruning and speedup..."
echo
"testing l1filter pruning and speedup..."
python3 basic_pruners_torch.py
--pruner
l1filter
--pretrain-epochs
1
--fine-tune-epochs
1
--model
vgg16
--dataset
cifar10
--pretrained-model-dir
experiment_data/pretrain_cifar10_vgg16.pth
python3 basic_pruners_torch.py
--pruner
l1filter
--pretrain-epochs
1
--fine-tune-epochs
1
--model
vgg16
--dataset
cifar10
--pretrained-model-dir
experiment_data/pretrain_cifar10_vgg16.pth
python3 model_speedup.py
--example_name
l1filter
python3
speedup/
model_speedup.py
--example_name
l1filter
echo
"testing apoz pruning and speedup..."
echo
"testing apoz pruning and speedup..."
python3 basic_pruners_torch.py
--pruner
apoz
--pretrain-epochs
1
--fine-tune-epochs
1
--model
vgg16
--dataset
cifar10
--pretrained-model-dir
experiment_data/pretrain_cifar10_vgg16.pth
python3 basic_pruners_torch.py
--pruner
apoz
--pretrain-epochs
1
--fine-tune-epochs
1
--model
vgg16
--dataset
cifar10
--pretrained-model-dir
experiment_data/pretrain_cifar10_vgg16.pth
python3 model_speedup.py
--example_name
apoz
python3
speedup/
model_speedup.py
--example_name
apoz
echo
'testing level pruner pruning'
echo
'testing level pruner pruning'
python3 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
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment