from local

This commit is contained in:
dohe0342 2023-03-30 00:16:14 +09:00
parent 018c194952
commit d7e60cad42
2 changed files with 3 additions and 1 deletions

View File

@ -1080,7 +1080,8 @@ def train_one_epoch(
if params.print_diagnostics and batch_idx == 5:
return
'''
if (
rank == 0
and params.batch_idx_train > 0
@ -1091,6 +1092,7 @@ def train_one_epoch(
model_cur=model,
model_avg=model_avg,
)
'''
if (
params.batch_idx_train > 0