原因: 数据没有给cuda 解决办法
img_patch = torch.tensor(np.expand_dims(img_arr, axis=[0,1])).to(device)
这样就可以啦
上一篇 C++内存管理
下一篇 深度神经网络对人工智能推动的发展评述与应用分析
版权所有 (c)2021-2022 MSHXW.COM
ICP备案号:晋ICP备2021003244-6号