From a9e5d9e5adbb674a5e4b71e0b47084259e81bb90 Mon Sep 17 00:00:00 2001 From: Aman Arora Date: Thu, 8 Apr 2021 03:41:40 -0400 Subject: [PATCH] log loss as before --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 631815ac..7c6f9d4b 100755 --- a/train.py +++ b/train.py @@ -723,7 +723,7 @@ def train_one_epoch( if hasattr(optimizer, 'sync_lookahead'): optimizer.sync_lookahead() - return OrderedDict([('train_loss', losses_m.avg)]) + return OrderedDict([('loss', losses_m.avg)]) def validate(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix=''): @@ -785,7 +785,7 @@ def validate(model, loader, loss_fn, args, amp_autocast=suppress, log_suffix='') log_name, batch_idx, last_idx, batch_time=batch_time_m, loss=losses_m, top1=top1_m, top5=top5_m)) - metrics = OrderedDict([('val_loss', losses_m.avg), ('top1', top1_m.avg), ('top5', top5_m.avg)]) + metrics = OrderedDict([('loss', losses_m.avg), ('top1', top1_m.avg), ('top5', top5_m.avg)]) return metrics