Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Changing the configuration of output_dir having no effect in Databricks. #277

Open
hv129 opened this issue Jan 19, 2024 · 1 comment
Open

Comments

@hv129
Copy link

hv129 commented Jan 19, 2024

finetuning_config = model.finetuning_config()
finetuning_config.output_dir = "model_cp/"

Changing the model's output directory as above (mentioned in https://xturing.stochastic.ai/configuration/finetune_configure/ ).
This is having no effect in Databricks and the model checkpoints are still getting stored in the default location "saved_model".

@hv129
Copy link
Author

hv129 commented Jan 22, 2024

The output paths seem to be hardcoded inside the file xturing.trainers.lightning_trainer.py

line 27: saved_path: str = "saved_model"
line 113: checkpoints_dir_path = Path("saved_model")

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant