trainer resource as argument

guo-yong-zhi · Jul 17, 2021 · 64d25a7 · 64d25a7 · guo-yong-zhi · Jul 18, 2021
1 parent e7d3fc9
commit 64d25a7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "Stuffing"
 uuid = "4175e07e-e5b7-423e-8796-3ea7f6d48281"
 authors = ["guoyongzhi <guo-yong-zhi@outlook.com>"]
-version = "0.4.0"
+version = "0.4.1"
 
 [deps]
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"

diff --git a/src/train.jl b/src/train.jl
@@ -469,7 +469,7 @@ end
 
 function train!(ts, nepoch::Number=-1, args...; 
         trainer=trainepoch_EM2!, patient::Number=trainer(:patient), optimiser=Momentum(η=1/4, ρ=0.5), 
-        callbackstep=1, callbackfun=x->x, teleporting=i->true, kargs...)
+    callbackstep=1, callbackfun=x->x, teleporting=i->true, resource=trainer(inputs=ts), kargs...)
     teleporton = true
     if teleporting isa Function
         on = teleporting
@@ -495,7 +495,6 @@ function train!(ts, nepoch::Number=-1, args...;
     nc_min_g = typemax(Int)
     teleport_count = 0.
     last_cinds = nothing
-    resource = trainer(inputs=ts)
     collpool = nothing
     if :collpool in keys(resource)
         collpool = resource[:collpool]