From 9c3082ad7fcf4a409beb7662092da3c84ce88914 Mon Sep 17 00:00:00 2001 From: txh18 Date: Mon, 28 Dec 2015 20:04:15 +0800 Subject: ... --- nerv/tnn/tnn.lua | 1 + 1 file changed, 1 insertion(+) 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 = {} -- cgit v1.2.3