6f539ad2324ddccc18b6ac39e4a2f4aeb12173a2,tests/test_layers_spatial_transformer.py,,,#,30

Before Change

net.print_params(False)

shape = s.outputs.get_shape().as_list()
if shape[1:] != [40, 40, 1]:
    raise Exception("shape do not match")

if len(net.all_layers) != 9:
    raise Exception("layers do not match")

if len(net.all_params) != 12:
    raise Exception("params do not match")

if net.count_params() != 1667980:
    raise Exception("params do not match")

After Change

        self.assertEqual(self.net_n_params, 1667980)


if __name__ == "__main__":

    // tf.logging.set_verbosity(tf.logging.INFO)
    tf.logging.set_verbosity(tf.logging.DEBUG)

    unittest.main()

Instances