-
Notifications
You must be signed in to change notification settings - Fork 23
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
START FEATURE EXTRACTION出现错误 #10
Comments
我未遇到这种错误,需要更详细说明下环境:显卡,CUDA, python, pytorch 版本 |
@Tonytom66 谢谢,应该是1050显存只有2GB,显存不够。参考eriklindernoren/PyTorch-YOLOv3#59 |
@Tonytom66 请尝试在python video_backbone/TSP/extract_features/extract_features.py 指令后面多加一个参数 --batch-size 这里batch size默认为32,减小batch size后应该可以解决此问题。 |
@ttengwang 这确实解决了我的问题,非常感谢!!! |
作者您好,我运行您的代码之后,在START FEATURE EXTRACTION阶段出现RuntimeError: CUDA error: unknown error,其它地方均无错误,请问我该如何解决它?
![image](https://user-images.githubusercontent.com/50938317/144397629-66a00b3c-c6d3-4a5f-9ece-741e737a586d.png)
![image](https://user-images.githubusercontent.com/50938317/144397698-32ec1c93-0fac-4494-863b-57f35bfb4014.png)
The text was updated successfully, but these errors were encountered: