diff --git a/code/rnnrbm.py b/code/rnnrbm.py index 34e85caf..8016d555 100644 --- a/code/rnnrbm.py +++ b/code/rnnrbm.py @@ -196,7 +196,7 @@ def __init__(self, n_hidden=150, n_hidden_recurrent=100, lr=0.001, n_hidden_recurrent) gradient = T.grad(cost, params, consider_constant=[v_sample]) - updates_train.update(dict((p, p - lr * g) for p, g in zip(params, + updates_train.update(((p, p - lr * g) for p, g in zip(params, gradient))) self.train_function = theano.function([v], monitor, updates=updates_train)