loss 爲負數 pytorch

網絡中的:

self.softmax = nn.Softmax(dim=1)
改爲:

self.softmax = nn.LogSoftmax(dim=1)
 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章