From 00ec15861dc6a062855a96b2d4c04f20d75662d9 Mon Sep 17 00:00:00 2001 From: chende Date: Tue, 28 May 2024 09:54:00 +0000 Subject: [PATCH] fix resnet50 loss print error --- Vision/classification/image/resnet50/train.py | 1 + 1 file changed, 1 insertion(+) diff --git a/Vision/classification/image/resnet50/train.py b/Vision/classification/image/resnet50/train.py index c1ba49ba4..37858c4e1 100644 --- a/Vision/classification/image/resnet50/train.py +++ b/Vision/classification/image/resnet50/train.py @@ -257,6 +257,7 @@ def train_one_epoch(self): else: top1_acc = 0 + loss = loss * self.world_size self.meter_train_iter(loss, top1_acc) self.cur_batch += 1