fd723d719ddb49b66a5084215343de763a40083d,trainer.py,Trainer,train_one_epoch,#Trainer#,154

Before Change

            // print to screen
            if i % self.print_freq == 0:
                print(
                    "Epoch: [{0}][{1}/{2}]\t"
                    "Time: {batch_time.val:.3f} ({batch_time.avg:.3f})\t"
                    "Train Loss: {loss.val:.4f} ({loss.avg:.4f})\t"
                    "Train Acc: {acc.val:.3f} ({acc.avg:.3f})".format(
                        epoch, i, len(self.train_loader),
                        batch_time=batch_time, loss=losses, acc=accs)
                )

        // log to tensorboard
        if self.use_tensorboard:
            log_value("train_loss", losses.avg, epoch)

After Change

                pbar.set_description(
                    (
                        "{:.1f}s - loss: {:.3f} - acc: {:.3f}".format(
                            (toc-tic), loss.data[0], acc.data[0]
                            )
                    )
                )
                pbar.update(self.batch_size)

            // log to tensorboard
            if self.use_tensorboard:
                log_value("train_loss", losses.avg, epoch)

Instances