andrewgordonwilson / bayesgan

Tensorflow code for the Bayesian GAN (https://arxiv.org/abs/1705.09558) (NIPS 2017)
Other
1.02k stars 175 forks source link

run_bgan does not actually switch to use specified optimizer #11

Open julie-jiang opened 6 years ago

julie-jiang commented 6 years ago

In lines 67-70:

if train_iter == 5000:
            print "Switching to user-specified optimizer"
            optimizer_dict = {"disc": dcgan.d_optims_adam,
                                          "gen": dcgan.g_optims_adam}

I believe what you intend to have here is dcgan.d_optims and dcgan.g_optims

Xiaohui9607 commented 5 years ago

I believe so