From db7e15c46fdddb3e152b36a53cba6f92fca5fd5a Mon Sep 17 00:00:00 2001
From: chenyu <chenyu@fastmail.com>
Date: Wed, 8 May 2024 15:14:41 -0400
Subject: [PATCH] hotfix resnet only log epoch start with RUNMLPERF (#4477)

---
 examples/mlperf/model_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/mlperf/model_train.py b/examples/mlperf/model_train.py
index 6f5f82082b..696450b3f8 100644
--- a/examples/mlperf/model_train.py
+++ b/examples/mlperf/model_train.py
@@ -184,7 +184,7 @@ def train_resnet():
   step_times = []
   for e in range(start_epoch, epochs):
     # ** train loop **
-    if MLLOGGER:
+    if MLLOGGER and RUNMLPERF:
       MLLOGGER.start(key=mllog_constants.EPOCH_START, value=e+1, metadata=dict(epoch_num=e+1))
     Tensor.training = True
     BEAM.value = TRAIN_BEAM