我想在使用LSTM模型时限制梯度值。
我使用optimizer.compute_gradients()
计算梯度时遇到了错误。以下是我的代码:
self.optimizer = tf.train.GradientDescentOptimizer(self.lr).minimize(self.loss)gvs = self.optimizer.compute_gradients(self.loss)
错误信息:
‘Operation’对象没有属性’compute_gradients’
如何解决这个问题,或者在tensorflow中是否有其他方法可以限制值?
回答:
尝试这样做:
self.optimizer = tf.train.GradientDescentOptimizer(self.lr)self.train_step = self.optimizer.minimize(self.loss)gvs = self.optimizer.compute_gradients(self.loss)