9a221ae50830159492ba6846f9680f2cfff8c552,hypergan/gan.py,GAN,train,#GAN#,64

Before Change

            self.sess = tf.Session(config=tf.ConfigProto())

    def train(self):
        return self.config["trainer.train"](self)

    def save(self, save_file):
        saver = tf.train.Saver()
        saver.save(self.sess, save_file)

After Change

            self.sess = tf.Session(config=tf.ConfigProto())

    def train(self):
        trainer = hc.Config(hc.lookup_functions(self.config.trainer))
        return trainer.run(self)

    def save(self, save_file):
        saver = tf.train.Saver()

Instances