Question

【iDesktopX】【11.0.1】使用影像分析的模型训练时报错：RuntimeEror:CUDA out of memory

Answer 1

【问题原因】单步运算量(batch size)设置过大 【解决办法】batch size这个值设置小一点，根据自己的显存来设置。比如显存为12G的机器，该值可以设置在10以内。