bcfb6021f2fd9b4412625ef7ee6bc703f40468b9,hypergan/gans/base_gan.py,BaseGAN,trainable_vars,#BaseGAN#,145

Before Change

    def trainable_vars(self):
        d_vars = list(set(self.d_vars()).intersection(tf.trainable_variables()))
        g_vars = list(set(self.g_vars()).intersection(tf.trainable_variables()))
        return d_vars, g_vars

    def save(self, save_file):
        print("[hypergan] Saving network to ", save_file)
        os.makedirs(os.path.expanduser(os.path.dirname(save_file)), exist_ok=True)

After Change

        return self.discriminator.variables()

    def trainable_vars(self):
        return self.trainable_d_vars(), self.trainable_g_vars()

    def trainable_d_vars(self):
        return list(set(self.d_vars()).intersection(tf.trainable_variables()))

Instances