from local

This commit is contained in:
dohe0342 2023-02-03 01:31:15 +09:00
parent 0b11d661be
commit 8fe06b88e7
2 changed files with 1 additions and 0 deletions

View File

@ -573,6 +573,7 @@ def train_one_epoch(
graph_compiler=graph_compiler, graph_compiler=graph_compiler,
valid_dl=valid_dl, valid_dl=valid_dl,
world_size=world_size, world_size=world_size,
wb=wb,
) )
model.train() model.train()
logging.info(f"Epoch {params.cur_epoch}, validation: {valid_info}") logging.info(f"Epoch {params.cur_epoch}, validation: {valid_info}")