fix bug in train

805094591 · Feb 16, 2019 · 3e58874 · 3e58874
1 parent 57b9385
commit 3e58874
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/tools/simnet/train/paddle/layers/paddle_layers.py b/tools/simnet/train/paddle/layers/paddle_layers.py
@@ -96,7 +96,9 @@ def ops(self, input):
         """
         operation
         """
-        proj = fluid.layers.fc(input=input, size=self.gru_dim * 3)
+        proj = fluid.layers.fc(input=input, size=self.gru_dim * 3
+                               param_attr=attr.ParamAttr(name="%s_fc.w" % self.name),
+                               bias_attr=attr.ParamAttr(name="%s_fc.b" % self.name))
         gru = fluid.layers.dynamic_gru(input=proj, size=self.gru_dim, 
                                        param_attr=attr.ParamAttr(name="%s.w" % self.name), 
                                        bias_attr=attr.ParamAttr(name="%s.b" % self.name))
@@ -118,7 +120,9 @@ def ops(self, input):
         """
         operation
         """
-        proj = fluid.layers.fc(input=input, size=self.lstm_dim * 4)
+        proj = fluid.layers.fc(input=input, size=self.lstm_dim * 4
+                               param_attr=attr.ParamAttr(name="%s_fc.w" % self.name),
+                               bias_attr=attr.ParamAttr(name="%s_fc.b" % self.name))
         lstm, _ = fluid.layers.dynamic_lstm(input=proj, size=self.lstm_dim * 4, 
                                             param_attr=attr.ParamAttr(name="%s.w" % self.name), 
                                             bias_attr=attr.ParamAttr(name="%s.b" % self.name))

diff --git a/tools/simnet/train/paddle/paddle_simnet.py b/tools/simnet/train/paddle/paddle_simnet.py
@@ -87,7 +87,7 @@ def train(conf_dict):
     # operate Optimization
     optimizer.ops(avg_cost)
     # optimize memory 
-    fluid.memory_optimize(fluid.default_main_program())
+    # fluid.memory_optimize(fluid.default_main_program())
     executor = fluid.Executor(place)
     executor.run(fluid.default_startup_program())
     # Get and run executor