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