empty teleporting early break

guo-yong-zhi · Jul 4, 2021 · 4e7b6d5 · 4e7b6d5 · guo-yong-zhi · Jul 7, 2021
1 parent 98c8dec
commit 4e7b6d5
Show file tree

Hide file tree

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

diff --git a/src/train.jl b/src/train.jl
@@ -493,7 +493,7 @@ function train!(ts, nepoch::Number=-1, args...;
     nc_min_t = typemax(Int)
     count_g = 0
     nc_min_g = typemax(Int)
-    teleport_count = 0
+    teleport_count = 0.
     last_cinds = nothing
     resource = trainer(inputs=ts)
     collpool = nothing
@@ -528,14 +528,14 @@ function train!(ts, nepoch::Number=-1, args...;
             if length(cinds) > 0
                 nc_min_t = typemax(nc_min_t)
                 reset!.(optimiser, ts[cinds])
-                cinds_set = Set(cinds)
-                if last_cinds == cinds_set
-                    teleport_count += 1
-                else
-                    teleport_count = 0
-                end
-                last_cinds = cinds_set
             end
+            cinds_set = Set(cinds)
+            if last_cinds == cinds_set
+                teleport_count += length(cinds_set) > 0 ? 1 : 0.5
+            else
+                teleport_count = 0.
+            end
+            last_cinds = cinds_set
         end
         if ep%callbackstep == 0
             callbackfun(ep)