-
Notifications
You must be signed in to change notification settings - Fork 96
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
Speeds of (GPU 8x, 14x and yolov4dense) running on desktop GPU (RTX2080Ti) are same #7
Comments
Yes sure. Our work targets on mobile devices and includes the pruning and compiler two parts. The repo on github is only about pruning. The pruned model needs the mobile compiler support to inference or acceleration. The mobile compiler is not open sourced right now. |
This is a fairly common question. I pinned your issue. |
Thanks for your answer. So I will wait for Nvidia's update |
|
Hi @Wuqiman , mobile compiler is not based on Alibaba MNN. Please refer to README.md. The compiler source code is associated with our collaborator at William & Mary, and has joint IP related stuff. We cannot open source this part now. Sorry for the inconvenience. |
I run:
detect.py --weights 'weights/best14x-49.pt' -- img-size 512 --> runing time (11ms on RTX2080ti)
detect.py --weights 'weights/best8x-514.pt -- img-size 512 --> runing time (11ms on RTX2080ti)
detect.py --weights 'weights/yolov4dense.pt ' -- img-size 512 --> runing time (11ms on RTX2080ti)
But when using check_compression.py, I see that FLOPS of these weights is still good
I just pip install -U -r requirements.txt, without docker build
So can you explain to me about this problem?
The text was updated successfully, but these errors were encountered: