We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e88c885 commit 53163b0Copy full SHA for 53163b0
dsm/utilities.py
@@ -57,7 +57,7 @@ def pretrain_dsm(model, t_train, e_train, t_valid, e_valid,
57
risks=model.risks)
58
premodel.double()
59
60
- optimizer = get_optimizer(model, lr)
+ optimizer = get_optimizer(premodel, lr)
61
62
oldcost = float('inf')
63
patience = 0
0 commit comments