gc in training loop

2017-09-27 23:14:58 +01:00 · 2017-09-27 23:14:58 +01:00 · 7c8dba0b85
commit 7c8dba0b85
parent a32ae4914c
1 changed files with 1 additions and 0 deletions
--- a/src/optimise/train.jl
+++ b/src/optimise/train.jl
@ -13,5 +13,6 @@ function train!(m, data, opt; cb = () -> ())
    back!(l)
    opt()
    cb()
+    gc()
  end
 end