Does AMCPruner only support 'mobilenet' and 'mobilenetv2' currently?

[ichejun]

Hi!
Does AMCPruner only support 'mobilenet' and 'mobilenetv2' currently? Some comments in the code are confusing：

nni/nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py

Line 88 in 6098314

model type to prune, currently 'mobilenet' and 'mobilenetv2' are supported.

#2876
and seems it already support resnet?

nni/examples/model_compress/pruning/amc/amc_train.py

Line 16 in 6098314

from torchvision.models import resnet

These makes me confused, does AMCPruner support network structures other than 'mobilenet' and 'mobilenetv2' currently?

If yes, can I use it in more other network structures, or is there any restrictions of net when using the amc pruner?
If not, is the share index for residual connection here limits the expansion of the model? or is there any other restrictions?

nni/nni/algorithms/compression/pytorch/pruning/amc/channel_pruning_env.py

Line 387 in 6098314

if self.args.model_type == 'mobilenetv2': # TODO: to be tested! Share index for residual connection

Thanks~

[J-shang]

Yes, it supports resnet. You could pull requests to modify the comments if it is convenient. : )

Because AMCPruner develops based on mobilenet, mobilenetv2 open-source implementation, it is hard to expanse to other models.

[ichejun]

@J-shang OK, thanks~ I pull request to modify the comments in #3647, and going to try other nets maybe :)