Commit fd50cc1b authored by Vishnu Banna's avatar Vishnu Banna
Browse files

new lines

parent 3c62075e
...@@ -125,5 +125,6 @@ class DarknetTest(parameterized.TestCase, tf.test.TestCase): ...@@ -125,5 +125,6 @@ class DarknetTest(parameterized.TestCase, tf.test.TestCase):
# If the serialization was successful, the new config should match the old. # If the serialization was successful, the new config should match the old.
self.assertAllEqual(network.get_config(), new_network.get_config()) self.assertAllEqual(network.get_config(), new_network.get_config())
if __name__ == '__main__': if __name__ == '__main__':
tf.test.main() tf.test.main()
...@@ -149,5 +149,6 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): ...@@ -149,5 +149,6 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
decoder_from_config = decoders.YoloDecoder.from_config(config) decoder_from_config = decoders.YoloDecoder.from_config(config)
self.assertAllEqual(decoder.get_config(), decoder_from_config.get_config()) self.assertAllEqual(decoder.get_config(), decoder_from_config.get_config())
if __name__ == '__main__': if __name__ == '__main__':
tf.test.main() tf.test.main()
...@@ -69,5 +69,6 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase): ...@@ -69,5 +69,6 @@ class YoloDecoderTest(parameterized.TestCase, tf.test.TestCase):
head_from_config = heads.YoloHead.from_config(configs) head_from_config = heads.YoloHead.from_config(configs)
self.assertAllEqual(head.get_config(), head_from_config.get_config()) self.assertAllEqual(head.get_config(), head_from_config.get_config())
if __name__ == '__main__': if __name__ == '__main__':
tf.test.main() tf.test.main()
...@@ -383,5 +383,6 @@ class DarkRouteProcessTest(tf.test.TestCase, parameterized.TestCase): ...@@ -383,5 +383,6 @@ class DarkRouteProcessTest(tf.test.TestCase, parameterized.TestCase):
self.assertNotIn(None, grad) self.assertNotIn(None, grad)
return return
if __name__ == '__main__': if __name__ == '__main__':
tf.test.main() tf.test.main()
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment