CrossEntropyLoss()をBCEloss()に変更するとエラーが出る。
解決したいこと pytorchを使用して、画像の2値分類に取り組んでいます。 最近、2値分類の損失計算にはCrossEntropyLoss()ではなく、BCEloss()を使用したほうが良いこと...
Qiita is a knowledge sharing service for engineers.
- You can follow users and tags
- You can stock useful information
- You can make edit suggestions for articles