3d26748eb13f0f97a1c8e1b49292ad2c952b4018,sonnet/src/sgd_test.py,SGDTest,testNoneUpdate,#SGDTest#Any#,55

Before Change

  @parameterized.parameters(sgd.SGD, sgd.FastSGD)
  def testNoneUpdate(self, opt_class):
    parameters = [tf.Variable([1., 2.])]
    updates = [None]
    optimizer = opt_class(learning_rate=3.)
    optimizer.apply(updates, parameters)
    self.assertAllClose([[1., 2.]], [x.numpy() for x in parameters])

After Change

    updates = [None, tf.constant(3.)]
    optimizer = opt_class(learning_rate=3.)
    optimizer.apply(updates, parameters)
    self.assertAllClose(1., parameters[0].numpy())

  @parameterized.parameters(sgd.SGD, sgd.FastSGD)
  def testVariableLearningRate(self, opt_class):
    parameters = [tf.Variable([1., 2.]), tf.Variable([3., 4.])]

Instances