def test_get_train_op(self): """Tests get_train_op. """ var = tf.Variable(0.) loss = tf.nn.l2_loss(var) train_op = opt.get_train_op(loss) self.assertTrue(tf.contrib.framework.is_tensor(train_op))
def _get_train_op(self): train_op = opt.get_train_op( loss=self._pg_loss, variables=self._policy.trainable_variables, learning_rate=self._lr, hparams=self._hparams.optimization.todict()) return train_op
def _get_train_op(self, loss): varlist = collect_trainable_variables([ self._src_embedder, self._tgt_embedder, self._encoder, self._connector, self._decoder ]) return get_train_op(loss, variables=varlist, hparams=self._hparams.optimization)
def _get_train_op(self): train_op = opt.get_train_op( loss=tf.reduce_sum(self._td_error**2), variables=self._qnet.trainable_variables, hparams=self._hparams.optimization.todict()) return train_op