Skip to content

Commit 53163b0

Browse files
authored
Corrected utilities.py (#44)
In order to have the right model for pretrain.
1 parent e88c885 commit 53163b0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dsm/utilities.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def pretrain_dsm(model, t_train, e_train, t_valid, e_valid,
5757
risks=model.risks)
5858
premodel.double()
5959

60-
optimizer = get_optimizer(model, lr)
60+
optimizer = get_optimizer(premodel, lr)
6161

6262
oldcost = float('inf')
6363
patience = 0

0 commit comments

Comments
 (0)