Merge pull request #3747 from Nerei/master

[pycaffe] NetSpec: set tops by attribute or dictionary
BVLC · Feb 29, 2016 · e5315fe · e5315fe
2 parents c2769c1 + cb27776
commit e5315fe
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/python/caffe/net_spec.py b/python/caffe/net_spec.py
@@ -175,6 +175,12 @@ def __setattr__(self, name, value):
     def __getattr__(self, name):
         return self.tops[name]
 
+    def __setitem__(self, key, value):
+        self.__setattr__(key, value)
+
+    def __getitem__(self, item):
+        return self.__getattr__(item)
+
     def to_proto(self):
         names = {v: k for k, v in six.iteritems(self.tops)}
         autonames = Counter()