RuntimeError: CUDA error: an illegal memory access was encountered terminate called after throwing an instance of 'c10::Error'

PeterL1n / RobustVideoMatting

Robust Video Matting in PyTorch, TensorFlow, TensorFlow.js, ONNX, CoreML!

https://peterl1n.github.io/RobustVideoMatting/

GNU General Public License v3.0

8.61k stars 1.14k forks source link

Open 1234269 opened 3 years ago

1234269 commented 3 years ago

请问训练时有遇到这个bug吗

PeterL1n commented 3 years ago

没有

yuqilol commented 9 months ago

超出显卡内存了吧，workers调小一点