fix resnet50 loss print error

Oneflow-Inc · May 28, 2024 · 00ec158 · 00ec158
1 parent fc7cbf8
commit 00ec158
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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