TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gc in training loop

Browse Source
This commit is contained in:
Mike J Innes 2017-09-27 23:14:58 +01:00
parent a32ae4914c
commit 7c8dba0b85
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/optimise/train.jl
View File

@ -13,5 +13,6 @@ function train!(m, data, opt; cb = () -> ())
back!(l) back!(l)
opt() opt()
cb() cb()
gc()
end end
end end