from local
This commit is contained in:
parent
0b11d661be
commit
8fe06b88e7
Binary file not shown.
@ -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}")
|
||||||
|
|||||||
Reference in New Issue
Block a user