除錯報錯
runtimeerror: input type (torch.cuda.floattensor) and weight type (torch.floattensor) should be the same
截圖
說明
在除錯執行導師給的程式residual net的gunpoint程式時，出現報錯，除錯結果為輸入與網路不符。
原因
解決方案
net = mdl.resnet()
改為↓net = mdl.resnet().cuda()
環境
vscode+pytorch環境（python3.6.10）
pytorch使用記錄
1.resnet需要載入預訓練模型，但是最後一層fc層的輸出數不是1000。解決方案 先載入預訓練模型，然後取除了最後一層的所有層，再加一層自己需要的在後面。2.model.children 和model.modules 的區別。model.modules 和model.children 均為迭代器...
Pytorch學習記錄筆記
輸出 函式會返回兩個tensor，第乙個tensor是每行的最大值，softmax的輸出中最大的是1，所以第乙個tensor是全1的tensor 第二個tensor是每行最大值的索引。predicted torch.max outputs.data,1 下劃線表示不在乎返回的第乙個tensor值。只...
pytorch的基礎記錄
detach torch.abs input,out none 計算輸入張量的每個元素的絕對值 torch.acos input,out none 返回乙個新張量，包含輸入張量每個元素的反余弦函式 torch.add input,value,out none 對輸入張量input逐元素加上標量值va...