START FEATURE EXTRACTION出现错误

ttengwang / PDVC

End-to-End Dense Video Captioning with Parallel Decoding (ICCV 2021)

MIT License

200 stars 23 forks source link

Closed panda-tang closed 2 years ago

panda-tang commented 2 years ago

作者您好，我运行您的代码之后，在START FEATURE EXTRACTION阶段出现RuntimeError: CUDA error: unknown error，其它地方均无错误，请问我该如何解决它？

ttengwang commented 2 years ago

我未遇到这种错误，需要更详细说明下环境：显卡，CUDA, python, pytorch 版本

panda-tang commented 2 years ago

我未遇到这种错误，需要更详细说明下环境：显卡，CUDA, python, pytorch 版本显卡：NVIDIA GeForce GTX 1050 CUDA Version 10.1.243 Python3.7 pytorch1.71

ttengwang commented 2 years ago

我未遇到这种错误，需要更详细说明下环境：显卡，CUDA, python, pytorch 版本显卡：NVIDIA GeForce GTX 1050 CUDA Version 10.1.243 Python3.7 pytorch1.71

@Tonytom66 谢谢，应该是1050显存只有2GB，显存不够。参考https://github.com/eriklindernoren/PyTorch-YOLOv3/issues/59

ttengwang commented 2 years ago

@Tonytom66 请尝试在python video_backbone/TSP/extract_features/extract_features.py 指令后面多加一个参数 --batch-size 这里batch size默认为32，减小batch size后应该可以解决此问题。

panda-tang commented 2 years ago

@ttengwang 这确实解决了我的问题，非常感谢！！！