from local

This commit is contained in:
dohe0342 2023-03-29 22:05:23 +09:00
parent ddff97fce6
commit 8efc363819
2 changed files with 2 additions and 0 deletions

View File

@ -1099,11 +1099,13 @@ def train_one_epoch(
rank=rank,
)
del params.cur_batch_idx
'''
remove_checkpoints(
out_dir=params.exp_dir,
topk=params.keep_last_k,
rank=rank,
)
'''
if batch_idx % 100 == 0 and params.use_fp16:
# If the grad scale was less than 1, try increasing it. The _growth_interval