From 432b9c32220ed7980012d7bccb3f10765ffeee21 Mon Sep 17 00:00:00 2001
From: Mike Innes <mike.j.innes@gmail.com>
Date: Tue, 6 Mar 2018 02:45:31 +0000
Subject: [PATCH] loadparams!

---
 src/treelike.jl | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/treelike.jl b/src/treelike.jl
index 4d896771..cac90e4e 100644
--- a/src/treelike.jl
+++ b/src/treelike.jl
@@ -40,6 +40,14 @@ end
 
 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(m) = mapleaves(x -> adapt(Array, x), m)