36093633b630d0934dae5b9abbda83d775f0e845,trainer/trainer.py,Trainer,_train_epoch,#Trainer#Any#,33

Before Change

                total_metrics[i] += metric(y_output, y_target)

            total_loss += loss.data[0]
            log_step = int(np.sqrt(self.batch_size))
            if self.verbosity >= 2 and batch_idx % log_step == 0:
                self.logger.info("Train Epoch: {} [{}/{} ({:.0f}%)] Loss: {:.6f}".format(
                    epoch, batch_idx * len(data), len(self.data_loader) * len(data),
                    100.0 * batch_idx / len(self.data_loader), loss.data[0]))

After Change

                    100.0 * batch_idx / len(self.data_loader), 
                    loss.data[0]))

        log = {
            "loss": total_loss / len(self.data_loader), 
            "metrics": (total_metrics / len(self.data_loader)).tolist()
        }

        if self.valid:
            val_log = self._valid_epoch()
            log = {**log, **val_log}

Instances