from local
This commit is contained in:
parent
b21b48c7ec
commit
44105cd893
Binary file not shown.
@ -1123,7 +1123,8 @@ def run(rank, world_size, args, wb=None):
|
|||||||
if params.print_diagnostics:
|
if params.print_diagnostics:
|
||||||
diagnostic.print_diagnostics()
|
diagnostic.print_diagnostics()
|
||||||
break
|
break
|
||||||
|
|
||||||
|
'''
|
||||||
save_checkpoint(
|
save_checkpoint(
|
||||||
params=params,
|
params=params,
|
||||||
model=model,
|
model=model,
|
||||||
@ -1134,6 +1135,7 @@ def run(rank, world_size, args, wb=None):
|
|||||||
scaler=scaler,
|
scaler=scaler,
|
||||||
rank=rank,
|
rank=rank,
|
||||||
)
|
)
|
||||||
|
'''
|
||||||
|
|
||||||
logging.info("Done!")
|
logging.info("Done!")
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user