new prompters, misc fixes for output dir missing using fsdp, and changing max seq len

This commit is contained in:
Wing Lian
2023-06-05 22:41:00 -04:00
parent 328c3bce96
commit 4ac9e251b7
4 changed files with 114 additions and 0 deletions

View File

@@ -279,6 +279,9 @@ def train(
logging.info(
f"Using Auto-resume functionality to start with checkpoint at {resume_from_checkpoint}"
)
if not Path(cfg.output_dir).is_dir():
os.makedirs(cfg.output_dir, exist_ok=True)
trainer.train(resume_from_checkpoint=resume_from_checkpoint)
logging.info(f"Training Completed!!! Saving pre-trained model to {cfg.output_dir}")