mit-han-lab / gan-compression

[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs
Other
1.1k stars 148 forks source link

How to distill a specific structure like "16_16_32_16_32_32_16_16"? #13

Closed MarSaKi closed 4 years ago

MarSaKi commented 4 years ago

I notice your scripts can distill resblock of 16, 24, 32. But how to distill a specific structure like "16_16_32_16_32_32_16_16"? I think your experiment in fig 6 of the paper, is it to compare 'pruning+distill" with "gan-compressIon" in the same MACs and resblock structure?