6aad1de658a933d3fa376f7fe9abf419da8a8bd2,python/baseline/tf/seq2seq/train.py,Seq2SeqTrainerTf,__init__,#Seq2SeqTrainerTf#Any#,16

Before Change

        self.test_loss = model.create_test_loss()
        self.model = model
        self.global_step, self.train_op = optimizer(self.loss, colocate_gradients_with_ops=True, **kwargs)
        self.log = logging.getLogger("baseline.timing")

    def checkpoint(self):
        self.model.saver.save(self.model.sess, "./tf-seq2seq-%d/seq2seq" % os.getpid(), global_step=self.global_step)

After Change

        self.test_loss = model.create_test_loss()
        self.model = model
        self.global_step, self.train_op = optimizer(self.loss, colocate_gradients_with_ops=True, **kwargs)
        self.nsteps = kwargs.get("nsteps", 500)

    def checkpoint(self):
        self.model.saver.save(self.model.sess, "./tf-seq2seq-%d/seq2seq" % os.getpid(), global_step=self.global_step)

Instances