Commit 111c9d39 authored by Kaushik Shivakumar's avatar Kaushik Shivakumar
Browse files

target assigner test

parent a6f36d27
...@@ -2280,5 +2280,4 @@ class DETRTargetAssignerTest(test_case.TestCase): ...@@ -2280,5 +2280,4 @@ class DETRTargetAssignerTest(test_case.TestCase):
self.assertEqual(reg_weights_out.dtype, np.float32) self.assertEqual(reg_weights_out.dtype, np.float32)
if __name__ == '__main__': if __name__ == '__main__':
tf.enable_v2_behavior()
tf.test.main() tf.test.main()
\ No newline at end of file
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