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

loadparams!

Browse Source
This commit is contained in:
Mike Innes 2018-03-06 02:45:31 +00:00
parent 6f0d5a44ec
commit 432b9c3222
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/treelike.jl
View File

@ -40,6 +40,14 @@ end
params(m...) = params(m) params(m...) = params(m)
function loadparams!(m, xs)
for (p, x) in zip(params(m), xs)
size(p) == size(x) ||
error("Expected param size $(size(p)), got $(size(x))")
copy!(data(p), data(x))
end
end
# CPU/GPU movement conveniences # CPU/GPU movement conveniences
cpu(m) = mapleaves(x -> adapt(Array, x), m) cpu(m) = mapleaves(x -> adapt(Array, x), m)