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,
|
||||
valid_dl=valid_dl,
|
||||
world_size=world_size,
|
||||
wb=wb,
|
||||
)
|
||||
model.train()
|
||||
logging.info(f"Epoch {params.cur_epoch}, validation: {valid_info}")
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user