tenstorrent / pytorch2.0_ttnn

⭐️ TTNN Compiler for PyTorch 2.0 ⭐️ It enables running PyTorch2.0 models on Tenstorrent hardware
https://tenstorrent.github.io/tt-metal/latest/ttnn/
25 stars 5 forks source link

Add model YOLOv5. #139

Closed cthsieh closed 1 month ago

cthsieh commented 2 months ago

Ticket

None.

Problem description

Add model YOLOv5 to trace.

This test depends on fixture manage_depedencies which was developed in PR 137. So, review PR 137 first, and then I continue to finish this PR in case of any change according to PR 137.

What's changed

Add model YOLOv5.