Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#270 Support for reading pretrained YoloX models with old blocks

Merged
Ofri Masad merged 1 commits into Deci-AI:master from deci-ai:feature/SG-193_yolox_old_ckpts_support
1 changed files with 0 additions and 3 deletions
  1. 0
    3
      tests/unit_tests/yolox_unit_test.py
@@ -39,9 +39,6 @@ class TestYOLOX(unittest.TestCase):
                 output_augment = yolo_model(dummy_input)
                 output_augment = yolo_model(dummy_input)
                 self.assertIsNotNone(output_augment)
                 self.assertIsNotNone(output_augment)
 
 
-                # MODEL IS ANCHORLESS
-                self.assertTrue(yolo_model._head.anchors.num_anchors == 0)
-
 
 
 if __name__ == '__main__':
 if __name__ == '__main__':
     unittest.main()
     unittest.main()
Discard