search

li-plus / DSNet

DSNet: A Flexible Detect-to-Summarize Network for Video Summarization
https://ieeexplore.ieee.org/document/9275314
MIT License
209 stars 50 forks source link

While running the infer.py file for the custom video summarization I am getting an error #9

Closed Akash-Kunwar closed 3 years ago

Akash-Kunwar commented 3 years ago

Loading DSNet model ... Preprocessing source video ...

ERROR

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=383 error=11 : invalid argument

Predicting summary ... Writing summary video ...

li-plus commented 3 years ago

It seems to be a CUDA error. I google this error and find this issue https://github.com/pytorch/pytorch/issues/15797. Does it help you?

Akash-Kunwar commented 3 years ago

Yes, I saw that issue but for me running the infer.py script again helped to resolve the error Thanks for helping