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

fix params

Browse Source
This commit is contained in:
Iblis Lin 2018-02-08 12:55:16 +00:00 committed by Mike J Innes
parent 4511936a87
commit f7fdfbe3a9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/treelike.jl
View File

@ -33,9 +33,11 @@ function prefor(f, x; seen = OSet())
return
end
using Flux.Tracker: istracked
function params(m)
ps = []
prefor(p -> p isa TrackedArray && push!(ps, p), m)
prefor(p -> istracked(p) && push!(ps, p), m)
return ps
end