-
Notifications
You must be signed in to change notification settings - Fork 66
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
Implement ttnn yolov4 with new conv API #10270
Comments
#10068 contains the implementation of ttnn yolov4 with new conv API. In head sub-module, we are using torch conv in three places as out_channels=255 is not working fine(#10256). While running yolov4 module we are facing OOM issue in DS5 sub_module first conv itself, But while running DS5 separately works fine. Also in DS1 to DS5 we are using relu instead mish activation due to following issue in ttnn.mish(#9877). |
All the sub-modules and yolov4 model works fine, the YOLOv4 model produces three outputs, and their PCC are greater than 0.95, 0.96, and 0.98, respectively. Conv issue : 10256 Currently three fallback conv is used and three maxpool operations are used in the pipeline. |
Closing this issue, since yolov4 is implemented using new conv-api and merged to main. |
Implement ttnn yolov4 and demo for yolov4 with new conv API(ttnn.conv2d).
The text was updated successfully, but these errors were encountered: