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

7add1c69 · M_bones · GitHub · dddf0b9e · 7add1c69
Unverified Commit 7add1c69 authored May 17, 2021 by M_bones Committed by GitHub May 17, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py ...ms/compression/pytorch/pruning/amc/channel_pruning_env.py +1 -1

No files found.
--- 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