From 7c8dba0b85d1cd8e114ce402879d252d69767876 Mon Sep 17 00:00:00 2001
From: Mike J Innes <mike.j.innes@gmail.com>
Date: Wed, 27 Sep 2017 23:14:58 +0100
Subject: [PATCH] gc in training loop

---
 src/optimise/train.jl | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/optimise/train.jl b/src/optimise/train.jl
index 0a91e978..4ecc7793 100644
--- 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