这里出现类似错误往往是因为,conv2d()函数的第一个参数,需要时一个tensor行的数据,然而却赋值成了not后给的类型。
我出现问题的原因是,中间有一个函数缺少返回值,因此会造成NoneType的错误。其余错误类似,只需把送给conv2d()的数据转换成tensor即可;
这里:numpy 和 tensor数据相互转换的方法:
1.numpy->tensor
data = torch.tensor(data)
2.tensor->numpy
非训练数据(训练结束后的tensor)的转换:
data1 = np.array(data1)
如果是训练过程中需要转换,则:
data_ = data.cpu().data.numpy()