Join GitHub today
GitHub is home to over 20 million developers working together to host and review code, manage projects, and build software together.
add layer_dict to the python interface #4347
Conversation
|
Sure, seems reasonable to me. Could you add some unit tests? |
|
I'd be glad to, but where? :) (Sorry, it's my first time contributing to caffe, and I can't seem to find the tests) |
|
Nevermind, I found them in python/caffe/test |
|
Here are the tests. To check the equality of the pointed layers, I only checked the types, but since the layers of the test network are all different types anyway, it seemed reasonable. |
shelhamer
added Python focus
labels
Jul 11, 2016
shelhamer
merged commit c510e33
into
BVLC:master
Feb 17, 2017
1 check passed
continuous-integration/travis-ci/pr
The Travis CI build passed
Details
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
nitnelave commentedJun 21, 2016
Similar to the blobs map, have a layer map.
This turns
net.layers[list(net._layer_names).index("ip2")]into
net.layer_dict["ip2"]