mirror of
https://github.com/tinygrad/tinygrad.git
synced 2026-06-13 00:15:35 +08:00
hotfix resnet only log epoch start with RUNMLPERF (#4477)
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user