b519cde79a37ac166bb2ac8e42134bb09d8c9e0f,model/base_model.py,BaseRunner,train,#BaseRunner#Any#Any#Any#Any#Any#Any#Any#,177

Before Change

        epoch_op = self.tensors["epoch"]
        epoch = sess.run(epoch_op)
        string = "starting from epoch {}/{}, {} batches per epoch.".format(epoch+1, num_epochs, num_iters_per_epoch)
        logging.info(string)
        print(string)
        while epoch < num_epochs:
            train_args = self._get_train_args(epoch)
            string = "epoch {}|".format(str(epoch+1).zfill(num_digits))
            train_data_set.reset()

After Change

                    writer.add_summary(summary, global_step)
                if progress:
                    pbar.update(iter_idx)
            if progress:
                pbar.finish()
            train_data_set.complete_epoch()

            assign_op = epoch_op.assign_add(1)
            _, epoch = sess.run([assign_op, epoch_op])

Instances