1. 程式人生 > >問題:TypeError: nll_loss() got an unexpected keyword argument 'reduction'

問題:TypeError: nll_loss() got an unexpected keyword argument 'reduction'

got an unexpected keyword argument  多半是pytorch版本問題

檢視原始檔,沒有reduction這個引數,但是官方文件裡有。

import torch
print(torch.__version__) 

輸出:0.4.0

解除安裝舊版本

pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl

安裝新版本 

pip uninstall torch

版本更新為0.4.1

成功執行