aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authortxh18 <[email protected]>2015-12-28 20:04:15 +0800
committertxh18 <[email protected]>2015-12-28 20:04:15 +0800
commit9c3082ad7fcf4a409beb7662092da3c84ce88914 (patch)
tree1d83e2d838a9d8be3f99556497e0d11f273684d3
parente59ce9cfcbe283955a616e0e05347d132d671e31 (diff)
...
-rw-r--r--nerv/tnn/tnn.lua1
1 files changed, 1 insertions, 0 deletions
diff --git a/nerv/tnn/tnn.lua b/nerv/tnn/tnn.lua
index 5351053..86b149e 100644
--- a/nerv/tnn/tnn.lua
+++ b/nerv/tnn/tnn.lua
@@ -218,6 +218,7 @@ function TNN:init(batch_size, chunk_size)
end
-- initialize sub layers
ref.layer:init(batch_size, chunk_size)
+ collectgarbage("collect")
end
local flags_now = {}