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

這裏出現相似錯誤每每是由於,conv2d()函數的第一個參數,須要時一個tensor行的數據,然而卻賦值成了not後給的類型。ide

我出現問題的緣由是,中間有一個函數缺乏返回值,所以會形成NoneType的錯誤。其他錯誤相似,只需把送給conv2d()的數據轉換成tensor便可;函數

這裏:numpy 和 tensor數據相互轉換的方法:it

1.numpy->tensorclass

data = torch.tensor(data)numpy

2.tensor->numpy方法

非訓練數據(訓練結束後的tensor)的轉換:cpu

data1 = np.array(data1)數據

若是是訓練過程當中須要轉換,則:di

data_ = data.cpu().data.numpy()view

相關文章
相關標籤/搜索