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.

Sign up for GitHub

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

Jump to bottom

ENH: Pass trained model along in TorchTrainer to make finetuning possible. #168

Open
justanotherariel opened this issue Jun 7, 2024 · 0 comments
Open

ENH: Pass trained model along in TorchTrainer to make finetuning possible. #168

justanotherariel opened this issue Jun 7, 2024 · 0 comments
Labels
01 - Enhancement New feature or request 11 - Trainer Issues related to model trainers needs milestone Marks an issue as needing a milestone

Comments

@justanotherariel
Copy link
Contributor

Right now, finetuning a previously trained model with TorchTrainer is not possible as the weights are not passed along. Come up with a way to pass weights along and continue training if pre-trained weights are present.
@justanotherariel justanotherariel added 01 - Enhancement New feature or request 11 - Trainer Issues related to model trainers labels Jun 7, 2024
@schobbejak schobbejak added the needs milestone Marks an issue as needing a milestone label Jun 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
01 - Enhancement New feature or request 11 - Trainer Issues related to model trainers needs milestone Marks an issue as needing a milestone
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@justanotherariel @schobbejak