Skip to content

Commit 1197464

Browse files
author
Ye Shaokai
committed
restoring the resume
1 parent b8bfb7c commit 1197464

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

llava/train/train.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1733,9 +1733,8 @@ def make_inputs_require_grad(module, input, output):
17331733

17341734

17351735
if list(pathlib.Path(training_args.output_dir).glob("checkpoint-*")):
1736-
#trainer.train(resume_from_checkpoint=True)
1737-
# for debug purpose, let's not resume
1738-
trainer.train()
1736+
trainer.train(resume_from_checkpoint=True)
1737+
#trainer.train()
17391738
else:
17401739
trainer.train()
17411740
trainer.save_state()

0 commit comments

Comments
 (0)