多GPU運行PyTorch報錯dimension specified as 0 but tensor has no dimensions

錯誤信息

dimension specified as 0 but tensor has no dimensions

問題原因

CrossEntropyLoss的輸入必須爲tensor，不能爲scalar（標量），即輸入的數據維度不能爲None.

解決辦法

使用view方法改變維度爲1，如target = target.view(-1),這樣假如原來target是1,使用後變爲[1],維度從None變爲1.

Problem Sovled.

參考

• https://blog.csdn.net/weixin_43380510/article/details/89173892