rename mod to module

liyangdev · Apr 12, 2016 · 94482b2 · 94482b2
1 parent 2253cfa
commit 94482b2
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/python/mxnet/__init__.py b/python/mxnet/__init__.py
@@ -44,6 +44,7 @@
 from . import torch
 from . import torch as th
 
-from . import mod
+from . import module
+from . import module as mod
 
 __version__ = base.__version__
diff --git a/python/mxnet/mod.py → python/mxnet/module.py b/python/mxnet/mod.py → python/mxnet/module.py
@@ -87,6 +87,10 @@ def backward(self):
         for texec in self.execs:
             texec.backward()
 
+    def update_metric(self, metric, labels):
+        for texec, islice in zip(self.execs, self.slices):
+            labels_slice = [label[islice] for label in labels]
+            metric.update(labels_slice, texec.outputs)
 
     def _bind_ith_exec(self, i, data_shapes, label_shapes, shared_group):
         data_shapes = self._sliced_shape(data_shapes, i)
@@ -182,9 +186,9 @@ def train(self, train_data, valid_data):
         pass
 
 
-class SymbolModule(BaseModule):
+class Module(BaseModule):
     def __init__(self, symbol, input_names, context=ctx.cpu(), work_load_list=None):
-        super(SymbolModule, self).__init__()
+        super(Module, self).__init__()
 
         if isinstance(context, ctx.Context):
             context = [context]
@@ -347,3 +351,6 @@ def update(self):
                            updater=self.updater,
                            num_device=len(self.context),
                            kvstore=self.kvstore)
+
+    def update_metric(self, metric, labels):
+        self.exec_group.update_metric(metric, labels)