search

mit-han-lab / gan-compression

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

bug in evolution.py #53

Closed artyomnaz closed 3 years ago

artyomnaz commented 3 years ago

Hello! Thank you all for your amazing work!

So, I think that your code in evolution.py might has an error. Why did you take sample2['channels'][1]] in line 124? Maybe you will consider sample2['channels'][i]]?

line 124: _newsample['channels'][i] = random.choice([sample1['channels'][i], sample2['channels'][1]])

lmxyy commented 3 years ago

Oh yes, you are right. Thanks for pointing it out!