這裏出現類似錯誤往往是因爲,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()