-
Notifications
You must be signed in to change notification settings - Fork 1.1k
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
Save the hyperparameters to the checkpoint #1012
Conversation
I wonder if we should add a
|
@rasbt It is important that this file contains exactly the arguments that were passed into the script, to capture exactly how it was called. It's not the same as meta data. It's not immediately clear to me how saving the path of the checkpoint where it is saved would be useful. Because that would become wrong once you move the checkpoint somewhere else. |
Ok that's fair. Never mind it was a silly idea. |
Saves a hyperparameters.yaml file to the checkpoint. Feel free to discuss the naming on this PR.
This will also be used to store the prompt template so that it can be dynamically loaded from the checkpoint after finetuning.
Part of #1005