loading from array

ai-ku · Nov 5, 2019 · 6ee43a6 · 6ee43a6
1 parent 36c4753
commit 6ee43a6
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/util.jl b/src/util.jl
@@ -76,7 +76,13 @@ end
 """
 function loadModel(fname::AbstractString)
     f = KnetLayers.load(fname)
-    return f["model"], f["opts"], f["vocab"], f["parser"]
+    prms,opts,vocab,parser = f["model"], f["opts"], f["vocab"], f["parser"]
+    ModelType = eval(Meta.parse(opts[:modelType]))
+    model = ModelType(opts,vocab)
+    for (wm,wl) in zip(params(model),prms)
+        copyto!(wm.value,wl)
+    end
+    return model,opts,vocab,parser
 end
 
 """