Skip to content

Commit 8f54f6d

Browse files
committed
change lr dtype
1 parent 5cbd7b0 commit 8f54f6d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: code/lstm.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def adadelta(lr, tparams, grads, x, mask, y, cost):
241241
242242
Parameters
243243
----------
244-
lr : float
244+
lr : Theano SharedVariable
245245
Initial learning rate
246246
tpramas: Theano SharedVariable
247247
Model parameters
@@ -303,7 +303,7 @@ def rmsprop(lr, tparams, grads, x, mask, y, cost):
303303
304304
Parameters
305305
----------
306-
lr : float
306+
lr : Theano SharedVariable
307307
Initial learning rate
308308
tpramas: Theano SharedVariable
309309
Model parameters

0 commit comments

Comments
 (0)