f7c759ca562303127a9991574d5a985d4dff99e8,sonnet/python/modules/gated_rnn_test.py,ConvLSTMTest,testTraining,#ConvLSTMTest#,996

Before Change

                                  initial_state=initial_state,
                                  dtype=tf.float32)

    loss = tf.reduce_mean(tf.square(output))
    train_op = tf.train.GradientDescentOptimizer(1).minimize(loss)
    init = tf.global_variables_initializer()
    with self.test_session() as sess:
      sess.run(init)
      sess.run(train_op)

  @parameterized.parameters(
      (snt.Conv1DLSTM, 1, False, 1, 1),
      (snt.Conv1DLSTM, 1, False, 1, 5),
      (snt.Conv1DLSTM, 1, False, 6, 1),

After Change

      return tf.reduce_mean(tf.square(output))

    train_op = tf.train.GradientDescentOptimizer(1).minimize(
        loss_fn if tf.executing_eagerly() else loss_fn())
    init = tf.global_variables_initializer()
    self.evaluate(init)
    self.evaluate(train_op)

  @parameterized.parameters(
      (snt.Conv1DLSTM, 1, False, 1, 1),
      (snt.Conv1DLSTM, 1, False, 1, 5),

Instances