原文:serWarning: Implicit dimension choice for softmax has been deprecated. Change the call to include dim=X as an argument.

torch版本的原因,torch . . 使用softmax時需要加上維度參數: p torch.nn.functional.softmax x,dim ...

2018-11-13 16:53 0 1085 推薦指數:

查看詳情

Can not squeeze dim[1], expected a dimension of 1

遇上如題所示問題,多半是因為使用了sparse_categorical_crossentropy作為損失函數 這會出現在分類問題中,這個激活函數要求如下: labels vect ...

Wed Feb 05 23:21:00 CST 2020 0 1053
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM