-
Notifications
You must be signed in to change notification settings - Fork 18.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
Save from python for net surgery #455
Conversation
As requested in #450. Thanks for the reminder to push this. |
I'll add an example notebook of doing a surgery and seeing the change in results, then let's merge. |
Rowland Depp did a similar task to convert imagenet pretrained models to fully convolutional, involving some late night ipython hacking - it's a great-to-have feature :) |
0. Scheme desired parameters. 1. Do surgery on the net through `net.params['name'][idx].data[...] = `. 2. Save post-operation net params by `net.save('fname')`. Handwoven deep nets, anyone?
@Yangqing I added a notebook detailing that very same model conversion. Merging now that an example of net surgery is included. |
Save from python for net surgery
Save from python for net surgery
net.params['name'][idx].data[...] =
.net.save('fname')
.Handwoven deep nets, anyone?
Surgeries done in this way: