diff --git a/src/optimise/train.jl b/src/optimise/train.jl index 4ecc7793..0a91e978 100644 --- a/src/optimise/train.jl +++ b/src/optimise/train.jl @@ -13,6 +13,5 @@ function train!(m, data, opt; cb = () -> ()) back!(l) opt() cb() - gc() end end