
エントリーの編集

エントリーの編集は全ユーザーに共通の機能です。
必ずガイドラインを一読の上ご利用ください。
Torchvisionのmodelの最後にsoftmaxがいらない件について。 - Qiita
記事へのコメント0件
- 注目コメント
- 新着コメント
このエントリーにコメントしてみましょう。
注目コメント算出アルゴリズムの一部にLINEヤフー株式会社の「建設的コメント順位付けモデルAPI」を使用しています

- バナー広告なし
- ミュート機能あり
- ダークモード搭載
関連記事
Torchvisionのmodelの最後にsoftmaxがいらない件について。 - Qiita
背景 以下のコードを見てくれればわかるが、最終層はLinerでsoftmax層が入っていない。 https://github.... 背景 以下のコードを見てくれればわかるが、最終層はLinerでsoftmax層が入っていない。 https://github.com/pytorch/vision/blob/master/torchvision/models/vgg.py 「え、これで大丈夫なの?」と思ったので、 理由 以下に全て書いてあった。 https://discuss.pytorch.org/t/torchvision-models-dont-have-softmax-layer/18071 学習時には、nn.CrossEntoropyLoss()を使うが、nn.LogSoftmaxとnn.NLLLoss.から成り立ってるので、必要ないとのこと。 推論する際も、各クラスの確率が欲しいならnn.functional.softmax()が必要だが、予測ならtorch.maxなどで最も大きい値のidxを抜いてこればいいと