from local

This commit is contained in:
dohe0342 2023-05-24 21:22:33 +09:00
parent 44c1a30255
commit c6c15b793a
3 changed files with 1 additions and 1 deletions

Binary file not shown.

View File

@ -1147,7 +1147,7 @@ def train_one_epoch(
if rank == 0:
for i, lora in enumerate(lora_modules):
lora.save_checkpoint(i, params.exp_dir)
lora.save_checkpoint(i, params.batch_idx_train, params.exp_dir)
'''
remove_checkpoints(