Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "61487e0a00f1956d413fa015884e8e95a9636b13"
Unverified Commit 7add1c69 authored by M_bones's avatar M_bones Committed by GitHub
Browse files

modify the comments of supporting resnet in channel_pruning_env.py (#3647)

parent dddf0b9e
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py
View file @ 7add1c69
...@@ -85,7 +85,7 @@ class ChannelPruningEnv: ...@@ -85,7 +85,7 @@ class ChannelPruningEnv:
args: args:
A Namespace object containing following arguments: A Namespace object containing following arguments:
model_type: str model_type: str
model type to prune, currently 'mobilenet' and 'mobilenetv2' are supported. model type to prune, currently 'mobilenet', 'mobilenetv2' and 'resnet' are supported.
flops_ratio: float flops_ratio: float
preserve flops ratio. preserve flops ratio.
lbound: float lbound: float
......
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