报错:
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_mm)
如果是用的网上预先下载好的权重,一定要先载入权重再将net改为cuda格式!!!
报错:
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument mat2 in method wrapper_mm)
如果是用的网上预先下载好的权重,一定要先载入权重再将net改为cuda格式!!!