Skip to content

Commit e5315fe

Browse files
committed
Merge pull request #3747 from Nerei/master
[pycaffe] NetSpec: set tops by attribute or dictionary
2 parents c2769c1 + cb27776 commit e5315fe

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

python/caffe/net_spec.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,12 @@ def __setattr__(self, name, value):
175175
def __getattr__(self, name):
176176
return self.tops[name]
177177

178+
def __setitem__(self, key, value):
179+
self.__setattr__(key, value)
180+
181+
def __getitem__(self, item):
182+
return self.__getattr__(item)
183+
178184
def to_proto(self):
179185
names = {v: k for k, v in six.iteritems(self.tops)}
180186
autonames = Counter()

0 commit comments

Comments
 (0)