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

Is it possible to re-train the model within glasspy? #7

Open
sarah-allec opened this issue Jun 27, 2023 · 2 comments
Open

Is it possible to re-train the model within glasspy? #7

sarah-allec opened this issue Jun 27, 2023 · 2 comments

Comments

@sarah-allec
Copy link

I'm interested in doing transfer learning from the original GlassNet model to the prediction of other glass properties - is this possible within the current glasspy framework?

@drcassar
Copy link
Owner

drcassar commented Jul 2, 2023

Hi @sarah-allec. Sure, that would be a great use of GlassNet! GlassNet is based on a lightning module, which itself is based on pytorch. You can perhaps freeze the weights and bias of the "hidden_layers", delete the "output_layer", and then add new layers to train for your transfer learning. I am planning on doing a tutorial on this, but I am not sure when it will be ready.

@sarah-allec
Copy link
Author

Thank you! I would greatly appreciate the tutorial but will attempt it on my own in the meantime.

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

2 participants