8648d437573e2e41b92b8fea13b4afe5e01851e8,baseline/tf/classify/training/eager.py,ClassifyTrainerEagerTf,_train,#ClassifyTrainerEagerTf#Any#Any#,76

Before Change

        for features, y in pg(loader):
            lossv = self.optimizer.update(self.model, features, y).numpy()
            batchsz = int(y.shape[0])
            report_lossv = lossv * batchsz
            epoch_loss += report_lossv
            epoch_div += batchsz
            self.nstep_agg += report_lossv

After Change

                )
                nstep_loss.reset_states()
                nstep_div.reset_states()
                self.nstep_start = time.time()

        epoch_loss = epoch_loss.result().numpy()
        epoch_div = epoch_div.result().numpy()
        metrics = self.calc_metrics(epoch_loss, epoch_div)

Instances