e61955b5553d98b4346817e0e9161f79fb23a40c,sonnet/python/modules/nets/alexnet_test.py,AlexNetTest,testNoDropoutInTesting,#AlexNetTest#,118

Before Change

    with self.test_session() as sess:
      sess.run(tf.global_variables_initializer())
      with self.assertRaisesRegexp(tf.errors.InvalidArgumentError, "keep_prob"):
        sess.run(output, feed_dict={inputs: np.random.rand(10, *input_shape),
                                    keep_prob: 0.7})
      // No exception if keep_prob=1
      sess.run(output, feed_dict={inputs: np.random.rand(10, *input_shape),
                                  keep_prob: 1.0})

  def testInputTooSmall(self):
    Check that an error is raised if the input image is too small.

After Change

    // No exception if keep_prob=1
    output = net(inputs, keep_prob=1.0, is_training=False)
    self.evaluate(output)

  def testInputTooSmall(self):
    Check that an error is raised if the input image is too small.

Instances