ScaledYOLOv4
ScaledYOLOv4 copied to clipboard
Terminate called after throwing an instance of 'c10::Error'
是在Scanning images环节出现了这个错误的, 如图:
我按照网上的指导把环境变量CUDA_LAUNCH_BLOCKING设为1,可是并没有用
我使用的GPU是三张NVIDIA GTX TITAN X,显存是12GB的
我要训练的模型是YOLOv4-P5,模型的参数量记得是70多M
请问这个错误是否是由于显存大小不够而导致的呢?