pytorch的圖像分類問題
前一段時間突然迷了，最後的全連接層（FC）輸出的維度對應的分類數目（Class_num）,這點是沒錯的，但是交叉熵損失函數公式裏邊是有預測概率的，必須由softmax得到，但是程序裏沒有看到softmax（），但是很多都會說到softmax層。後來經過查閱發現，解決辦法在交叉熵損失函數裏邊。 是由交叉熵損失函數的結構解決。
交叉熵損失函數：
鏈接: https://blog.csdn.net/c2250645962/article/details/106014693/?utm_medium=distribute.pc_relevant.none-task-blog-baidujs-1.