-
Notifications
You must be signed in to change notification settings - Fork 4.7k
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
#7329 load models in finetune mode core #7458
#7329 load models in finetune mode core #7458
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Haven't reviewed extensively yet, but an initial observation.
…oad_models_in_finetune_mode_core
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looking great 👍
meta["should_finetune"] = kwargs.get("should_finetune", False) | ||
if "epoch_override" in kwargs: | ||
meta[EPOCHS] = kwargs["epoch_override"] | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This needs a larger change which i'll address in my PR as part of making core policies finetunable. It's okay to keep as it is here.
Co-authored-by: Daksh Varshneya <d.varshneya@rasa.com>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
🚀
Co-authored-by: Tobias Wochinger <t.wochinger@rasa.com>
@joejuzl Can you add the required changes to custom policies in the migration guide as part of this PR itself? |
I think a constant for |
@dakshvar22 I think a separate PR for the migration guide would make more sense so we can start using this code easier in other PRs. |
Proposed changes:
Load the core models in fine-tuning mode.
Status (please check what you already did):
black
(please check Readme for instructions)