How do I compress the other GAN models

mit-han-lab / gan-compression

[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs

Other

1.1k stars 148 forks source link

Closed ypw1996 closed 2 years ago

ypw1996 commented 4 years ago

Hello, I want to use this method to compress the other GAN models, like starGAN, styleGAN. If it is feasible, What should I do?

jiaying96 commented 4 years ago

+1，which part should be modified to compress the other GAN models

lmxyy commented 4 years ago

There are several parts you may need to modify.

data/. You need to customize your dataset in this folder.
models/modules. You need to define your generator and discriminator pytorch modules here.
configs/. You need to define a searching space here.
supernets/. You need to create a "once-for-all" network trainer here. Specifically, you need to choose the matching layers and implement the training logic.

junyanz commented 4 years ago

Might be worth writing a tutorial on compressing other gans models.