From aa5d3962b9899860dfab8768ed7d6b60d192e546 Mon Sep 17 00:00:00 2001 From: "Dr. Kashif Rasul" Date: Tue, 17 Dec 2019 13:47:07 +0100 Subject: [PATCH] no validation for now --- pts/trainer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pts/trainer.py b/pts/trainer.py index 25ddf73..9803f7e 100644 --- a/pts/trainer.py +++ b/pts/trainer.py @@ -50,8 +50,7 @@ class Trainer: avg_epoch_loss += loss.item() it.set_postfix( ordered_dict={ - ("" if is_training else "validation_") - + "avg_epoch_loss": avg_epoch_loss/batch_no + "avg_epoch_loss": avg_epoch_loss/batch_no }, refresh=False, )