OverFeat icon indicating copy to clipboard operation
OverFeat copied to clipboard

Process for saving weights to file

Open nmante opened this issue 8 years ago • 2 comments

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++.

nmante avatar Jul 05 '16 23:07 nmante

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

chrisranderson avatar Jan 07 '17 17:01 chrisranderson

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.

nmante avatar Jan 10 '17 02:01 nmante