进行eval,加载模型到gpu时,出现错误:RuntimeError: CUDA error: out of memory
疑似原因:有一个相似的模型(网络结构名相同,通道数不同)在另一块gpu上训练着,导致加载错误
解决办法:先停止相似模型的训练
2024-01-13 17:27:40
进行eval,加载模型到gpu时,出现错误:RuntimeError: CUDA error: out of memory
疑似原因:有一个相似的模型(网络结构名相同,通道数不同)在另一块gpu上训练着,导致加载错误
解决办法:先停止相似模型的训练