set bind flag after bind completes (apache#12155)

leezu · Aug 16, 2018 · 85179e9 · 85179e9
1 parent db63a81
commit 85179e9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/mxnet/module/module.py b/python/mxnet/module/module.py
@@ -398,7 +398,6 @@ def bind(self, data_shapes, label_shapes=None, for_training=True,
 
         self.for_training = for_training
         self.inputs_need_grad = inputs_need_grad
-        self.binded = True
         self._grad_req = grad_req
 
         if not for_training:
@@ -454,6 +453,8 @@ def bind(self, data_shapes, label_shapes=None, for_training=True,
         if shared_module is not None and shared_module.optimizer_initialized:
             self.borrow_optimizer(shared_module)
 
+        self.binded = True
+
     def reshape(self, data_shapes, label_shapes=None):
         """Reshapes the module for new input shapes.