modified: dsm/utilities.py

autonlab · Jan 30, 2021 · 65ad4e2 · 65ad4e2
1 parent c9295ff
commit 65ad4e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dsm/utilities.py b/dsm/utilities.py
@@ -185,8 +185,8 @@ def train_dsm(model,
 
     if (costs[-1] >= oldcost) is True:
       if patience == 2:
-        maxm = np.argmax(costs)
-        model.load_state_dict(dics[maxm])
+        minm = np.argmin(costs)
+        model.load_state_dict(dics[minm])
 
         del dics
         gc.collect()