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