added wcost for biasparam in lm_trainer

1 files changed, 12 insertions, 1 deletions

diff --git a/nerv/examples/lmptb/tnn/layer_dag_t.lua b/nerv/examples/lmptb/tnn/layer_dag_t.lua
index cd5fba7..ade65cc 100644
--- a/nerv/examples/lmptb/tnn/layer_dag_t.lua
+++ b/nerv/examples/lmptb/tnn/layer_dag_t.lua
@@ -266,7 +266,9 @@ function DAGLayerT:set_inputs(input, t)
         end
         local layer = self.inputs[i][1]
         local port = self.inputs[i][2]
-        
+        if layer.inputs[t] == nil then
+            layer.inputs[t] = {}
+        end
         layer.inputs[t][port] = input[i]
     end
 end
@@ -278,6 +280,9 @@ function DAGLayerT:set_outputs(output, t)
         end
         local layer = self.outputs[i][1]
         local port = self.outputs[i][2]
+        if layer.outputs[t] == nil then
+            layer.outputs[t] = {}
+        end
         layer.outputs[t][port] = output[i]
     end
 end
@@ -286,6 +291,9 @@ function DAGLayerT:set_err_inputs(bp_err, t)
     for i = 1, #self.dim_out do
         local layer = self.outputs[i][1]
         local port = self.outputs[i][2]
+        if layer.err_inputs[t] == nil then
+            layer.err_inputs[t] = {}
+        end
         layer.err_inputs[t][port] = bp_err[i]
     end
 end
@@ -294,6 +302,9 @@ function DAGLayerT:set_err_outputs(next_bp_err, t)
     for i = 1, #self.dim_in do
         local layer = self.inputs[i][1]
         local port = self.inputs[i][2]
+        if layer.err_outputs[t] == nil then
+            layer.err_outputs[t] = {}
+        end 
         layer.err_outputs[t][port] = next_bp_err[i]
     end
 end