from local
This commit is contained in:
parent
223558f0c1
commit
3fca4c4221
Binary file not shown.
@ -691,6 +691,7 @@ def run(rank, world_size, args, wb=None):
|
|||||||
valid_dl=valid_dl,
|
valid_dl=valid_dl,
|
||||||
tb_writer=tb_writer,
|
tb_writer=tb_writer,
|
||||||
world_size=world_size,
|
world_size=world_size,
|
||||||
|
wb=wb,
|
||||||
)
|
)
|
||||||
|
|
||||||
save_checkpoint(
|
save_checkpoint(
|
||||||
|
|||||||
Reference in New Issue
Block a user