Corrected utilities.py (#44)

In order to have the right model for pretrain.
autonlab · Feb 13, 2021 · 53163b0 · 53163b0
1 parent e88c885
commit 53163b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dsm/utilities.py b/dsm/utilities.py
@@ -57,7 +57,7 @@ def pretrain_dsm(model, t_train, e_train, t_valid, e_valid,
                                        risks=model.risks)
   premodel.double()
 
-  optimizer = get_optimizer(model, lr)
+  optimizer = get_optimizer(premodel, lr)
 
   oldcost = float('inf')
   patience = 0