yolov6 running ERROR: torch._dynamo.exc.IncorrectUsage

Emily

• September 10, 2024 03:53

“ Error: torch._dynamo.exc.IncorrectUsage: disallow_in_graph is expected to be used on an already allowed callable (like torch.* ops). Allowed callables means callables that TorchDynamo puts as-is in the extracted graph. ”

I got above error while running YOLOv6. 

 The command used is “python tools/eval.py --specific-shape owlite --project yolo --baseline yolov6 ”.

I tried to create a baseline for YOLOv6-s in OwLite. I already searched other way and applied it on my code, but I can't solve above the problem.  

0

• 

  Sia

  • September 10, 2024 03:58

  Hi, 

  Could you please let me know the environment you're currently working on? (e.g. PyTorch version, OwLite version, etc..)

  Thanks,

  0
• 

  Emily

  • September 10, 2024 04:29

  I'm working on the following enviorment. 

  torch version: 2.4.0 / torchvision: 0.19.0/ OwLite version: 2.1.0

  1
• 

  Sia

  • September 10, 2024 09:08

  Hello,

  The error you are encountering (torch._dynamo.exc.IncorrectUsage: disallow_in_graph is expected to be used on an already allowed callable (like torch.* ops)) typically arises from compatibility issues between TorchDynamo and the versions of the libraries being used.

  In this case, the issue seems to be related to the specific versions of PyTorch and OwLite. Currently, OwLite (v2.1.0) supports PyTorch versions within the range torch>=2.1.2,<2.3. 

  To ensure compatibility, it is recommended to use OwLite version 2.1.0 together with PyTorch version 2.1.2.

  For further details regarding dependencies, please refer to the OwLite GitHub repository.

  Thanks!

  0
• 

  Emily

  • September 10, 2024 09:32

  Your advice was helpful! Thank you.

  0

Please sign in to leave a comment.