from local

This commit is contained in:
dohe0342 2023-04-18 14:09:09 +09:00
parent 000dc1af5c
commit bd35e337bc
2 changed files with 3 additions and 1 deletions

View File

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