diff options
Diffstat (limited to 'nerv/tnn/tnn.lua')
-rw-r--r-- | nerv/tnn/tnn.lua | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/nerv/tnn/tnn.lua b/nerv/tnn/tnn.lua index bcfeb40..7ae3172 100644 --- a/nerv/tnn/tnn.lua +++ b/nerv/tnn/tnn.lua @@ -466,7 +466,7 @@ function TNN:net_backpropagate(do_update) --propagate according to feeds_now local feeds_now = self.feeds_now for t = 1, self.chunk_size do --some layer maybe do not have outputs from time 1..chunk_size for id, ref in pairs(self.layers) do - self:backpropagate_dfs(ref, t) + self:backpropagate_dfs(ref, t, do_update) end end for t = 1, self.chunk_size do @@ -500,6 +500,9 @@ end --ref: the TNN_ref of a layer --t: the current time to propagate function TNN:backpropagate_dfs(ref, t, do_update) + if do_update == nil then + nerv.error("got a nil do_update") + end if self:out_of_feedrange(t) then return end |