NetSpec: allow setting blob names by string

MeJn · Feb 29, 2016 · cb27776 · cb27776
1 parent c2769c1
commit cb27776
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()