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

useful params method

Browse Source
This commit is contained in:
Mike J Innes 2017-11-07 19:34:35 +00:00
parent d6423eefe5
commit d4229c4815
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/treelike.jl
View File

@ -35,3 +35,5 @@ function params(m)
prefor(p -> p isa TrackedArray && push!(ps, p), m) prefor(p -> p isa TrackedArray && push!(ps, p), m)
return ps return ps
end end
params(m...) = params(m)