We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
Describe the bug Running benchmark with high loop_counts will cause the (shared) memory to fill up with, leading to a device hang.
loop_counts
Need to adjust the benchmark.py script to only store the input labels and ouputs for the first loop iteration into Python list objects.
benchmark.py
To Reproduce Steps to reproduce the behavior: python benchmark.py -d tt -m yolo_v5 -c s --task object_detection -mb 128 --loop_count 32 -opt 4 --save_output
python benchmark.py -d tt -m yolo_v5 -c s --task object_detection -mb 128 --loop_count 32 -opt 4 --save_output
The text was updated successfully, but these errors were encountered:
#22: Fix (shared) memory overfill causing device hang (#23)
b206678
milank94
Successfully merging a pull request may close this issue.
Describe the bug
Running benchmark with high
loop_counts
will cause the (shared) memory to fill up with, leading to a device hang.Need to adjust the
benchmark.py
script to only store the input labels and ouputs for the first loop iteration into Python list objects.To Reproduce
Steps to reproduce the behavior:
python benchmark.py -d tt -m yolo_v5 -c s --task object_detection -mb 128 --loop_count 32 -opt 4 --save_output
The text was updated successfully, but these errors were encountered: