diff options
-rw-r--r-- | nerv/tnn/tnn.lua | 1 |
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 = {} |