Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ttnn implementation for yolox_m #8591

Open
saichandax opened this issue May 17, 2024 · 2 comments
Open

ttnn implementation for yolox_m #8591

saichandax opened this issue May 17, 2024 · 2 comments
Assignees

Comments

@saichandax
Copy link
Contributor

reference implementation: https://github.com/Megvii-BaseDetection/YOLOX/tree/main

@ankitmcw
Copy link

ankitmcw commented May 17, 2024

Yolox_m current status

  1. Reference implementation of yolox_m completed
  2. Tested reference against torch: Getting pcc 1 in all three output tensors
  3. Reference vs torch graph comparison checked
  4. Currently working on ttnn implementation

Torch graph:
Yolox_m_torch.gv.pdf

Reference graph:
Yolox_m_reference.gv.pdf

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants