projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
91cd44b
)
Expose caffe.Net.params as an OrderedDict
author
Jonathan L Long
<jonlong@cs.berkeley.edu>
Tue, 11 Mar 2014 04:51:25 +0000
(21:51 -0700)
committer
Jonathan L Long
<jonlong@cs.berkeley.edu>
Sat, 15 Mar 2014 09:27:35 +0000
(
02:27
-0700)
python/caffe/pycaffe.py
patch
|
blob
|
history
diff --git
a/python/caffe/pycaffe.py
b/python/caffe/pycaffe.py
index
5d36489
..
8fbbcf9
100644
(file)
--- a/
python/caffe/pycaffe.py
+++ b/
python/caffe/pycaffe.py
@@
-15,6
+15,9
@@
class Net(CaffeNet):
super(Net, self).__init__(param_file, pretrained_param_file)
self._blobs = OrderedDict([(bl.name, bl)
for bl in super(Net, self).blobs])
+ self.params = OrderedDict([(lr.name, lr.blobs)
+ for lr in super(Net, self).layers
+ if len(lr.blobs) > 0])
@property
def blobs(self):