Closed yy2016417 closed 2 years ago
Hi @yy2016417 ,
Yes. Since most generative adversarial networks use Linear
and Conv2d
as their basic building blocks, you can apply NNI pruning and quantization tools to compress them. You can refer to here for examples of how to specify a config_list
and use the tools provided by NNI. Although we don't have an example for compressing GANs, the general process should be very similar.
As a side remark, I think only compressing the generator should be enough for most cases.
Hi @yy2016417 , Yes. Since most generative adversarial networks use
Linear
andConv2d
as their basic building blocks, you can apply NNI pruning and quantization tools to compress them. You can refer to here for examples of how to specify aconfig_list
and use the tools provided by NNI. Although we don't have an example for compressing GANs, the general process should be very similar. As a side remark, I think only compressing the generator should be enough for most cases.
Thanks for your help !
Thank you for a really great job! For GAN there are generator and discriminator.I would like to ask if it is possible to use NNI for model compression for this group of networks