devzhk / Implicit-Competitive-Regularization

Code for: Implicit Competitive Regularization in GANs
Apache License 2.0
114 stars 18 forks source link

cannot import name 'CGD' #2

Closed Johnson-yue closed 4 years ago

Johnson-yue commented 4 years ago

Hi, thank you for sharing your code. I will run train.ipynb not it is error

than I check file VisionData.py and optimizers.py I found:

In VisionData.py you import CGD module but in optimizers.py only have BCGD, OCGD, MCGD , also miss ACGD

devzhk commented 4 years ago

I forgot to remove the unnecessary import. I removed it just now. And yes, class name is a little confusing. I renamed the class to be consistent with the name in paper.

Thanks !