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

Process for saving weights to file #35

Open
nmante opened this issue Jul 5, 2016 · 2 comments
Open

Process for saving weights to file #35

nmante opened this issue Jul 5, 2016 · 2 comments

Comments

@nmante
Copy link

nmante commented Jul 5, 2016

How were the weights saved to disk? Were they extracted from Torch using getParameters then saved as a torch object using torch.save? I'm interested because I've trained my own smaller network and would like to load the weights from torch/lua training in C++.

@chrisranderson
Copy link

Maybe this will be helpful: https://github.com/jhjin/overfeat-torch

@nmante
Copy link
Author

nmante commented Jan 10, 2017

That's a nice option. For my purposes, I needed to load my own custom network (with it's own weights and structure). I ended up using https://github.com/mvitez/thnets to do this. It's a standalone library that let's you load, save and use custom networks directly in C++. I tweaked a bit to run on my machine (OS X). It worked for me, maybe it'll work for anyone else who sees this.

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