-
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
#10270: yolov4 implementation #10068
Conversation
5c45326
to
49fbdb0
Compare
The performance sheets of the following submodules are added herewith:
|
3892644
to
fa81d68
Compare
Performance sheet for neck sub-module : ops_perf_results_neck_2024_07_12_16_43_36.csv We are not able to generate perf sheets for head sub-module as the test hangs after completing the sub-module and also the yolov4 pipeline fails due to OOM issue in DS5 sub-module first conv itself ,but DS5 sub-module is passing when tested separately. Currently, Using torch maxpool in neck sub-module as ttnn.maxpool are making issues for those configuration #7746. |
196a75f
to
805d26a
Compare
Latest neck and head sub-module perf sheets: |
fdfea45
to
e6b824e
Compare
All the sub-modules work fine, the YOLOv4 model produces three outputs, and their PCC are greater than 0.95, 0.96, and 0.98, respectively. Performance sheets of all modules: Downsample1 sub_module : ops_perf_results_downsample1_17_2024_07_16_12_42_56.csv Downsample2 sub_module : ops_perf_results_downsample2_17_2024_07_16_12_44_45.csv Downsample3 sub_module : ops_perf_results_downsample3_17_2024_07_16_12_45_27.csv Downsample4 sub_module : ops_perf_results_downsample4_17_2024_07_16_12_46_15.csv Downsample15sub_module : ops_perf_results_downsample5_17_2024_07_16_12_46_55.csv Neck sub_module : ops_perf_results_neck_17_2024_07_16_12_48_17.csv Head sub_module : ops_perf_results_head_17_2024_07_16_12_47_31.csv yolov4 module : ops_perf_results_yolov4_17_2024_07_16_12_49_50.csv |
a5f91fd
to
edb0d05
Compare
Latest performance sheet, Below are the current FPS,
Note : Currently using 3 torch maxpool which is not visible in the perf sheets. |
edb0d05
to
20265bc
Compare
Latest perf sheet, Below are the current FPS,
|
7552b26
to
7abcdce
Compare
7abcdce
to
2fbdece
Compare
Closing this PR since it's outdated and the YOLOv4 model is in the main. |
This PR contains ttnn yolov4 implementation with new conv API.