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

…3647)
microsoft · May 17, 2021 · 7add1c6 · 7add1c6
1 parent dddf0b9
commit 7add1c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py b/nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py
@@ -85,7 +85,7 @@ class ChannelPruningEnv:
         args:
             A Namespace object containing following arguments:
             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
                 preserve flops ratio.
             lbound: float