load trained model if available

Author: ekinakyurek

configs/config2.jl

@@ -1,5 +1,5 @@
 o=Dict(:h5=>false,
-       :mfile=>nothing,
+       :mfile=>"train4_p2.jld2",
        :epochs=>30,
        :lr=>0.0001,
        :p=>12,

train.jl

@@ -14,8 +14,13 @@ trnqstns=nothing;
 valqstns=nothing;
 #MODEL
 #gpu(0)
-M    = MACNetwork(o);
-Mrun = MACNetwork(o);
+if o[:mfile] !=nothing && isfile(o[:mfile])
+    M,Mrun,o = loadmodel(o[:mfile])
+else
+    M    = MACNetwork(o);
+    Mrun = MACNetwork(o);
+end
+
 for (wr,wi) in zip(params(Mrun),params(M));
     wr.value[:] = wi.value[:]
 end