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
