shayneobrien / generative-models Goto Github PK View Code? Open in Web Editor NEW 493.0 17.0 75.0 146.27 MB Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN License: MIT License Jupyter Notebook 99.03% Python 0.97% python generative-adversarial-network machine-learning pytorch discriminator gan wasserstein vae generative-models began Introduction ยท People ยท Discuss
WGAN-GP mispelling In the notebook '04-wasserstein-gan-gradient-penalty.ipynb', the name of the model, 'WGANGP' and the trainer 'WGANGPTrainer' are incorrect concerning the corresponding src file (w_gp_gan.py). It should be changed into 'WGPGAN' and 'WGPGANTrainer'.
vae.py line 320 use this to plot replace these lines - plt.scatter(m1s, m2s, c=labels) plt.legend([str(i) for i in set(labels)]) with the lines below - scatter = plt.scatter(m1s, m2s, c=labels) plt.legend(handles=scatter.legend_elements()[0], labels=set(labels))
wgan optimizer hello :) In the W-GAN model you mention in the comments the use of RMSProp optimizer, but you use in the code the Adam optimizer.
Vue.js ๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
javascript JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Machine learning Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Facebook We are working to build community through open source technology. NB: members must have two-factor auth.