from local
This commit is contained in:
parent
ddff97fce6
commit
8efc363819
Binary file not shown.
@ -1099,11 +1099,13 @@ def train_one_epoch(
|
|||||||
rank=rank,
|
rank=rank,
|
||||||
)
|
)
|
||||||
del params.cur_batch_idx
|
del params.cur_batch_idx
|
||||||
|
'''
|
||||||
remove_checkpoints(
|
remove_checkpoints(
|
||||||
out_dir=params.exp_dir,
|
out_dir=params.exp_dir,
|
||||||
topk=params.keep_last_k,
|
topk=params.keep_last_k,
|
||||||
rank=rank,
|
rank=rank,
|
||||||
)
|
)
|
||||||
|
'''
|
||||||
|
|
||||||
if batch_idx % 100 == 0 and params.use_fp16:
|
if batch_idx % 100 == 0 and params.use_fp16:
|
||||||
# If the grad scale was less than 1, try increasing it. The _growth_interval
|
# If the grad scale was less than 1, try increasing it. The _growth_interval
|
||||||
|
|||||||
Reference in New Issue
Block a user