TypeError: conv2d(): argument 'input' (position 1) must be Tensor, not NoneType

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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章