...

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 = {}