pytorch中交叉熵的使用

關於pytorch中交叉熵的使用,pytorch的交叉熵是其loss function的一種且包含了softmax的過程。 pytorch中交叉熵函數是nn.CrossEntropyLoss()。其參數包括:weight,size_averaged,reduce weight參數通常默認值是0,如果你的訓練樣本很不均衡的話,可以設置其值。 size_averaged參數是設置是否取均值,默認是對b
相關文章
相關標籤/搜索