Why using BCELoss? That is for binary classification while we have 20 classes
lucasjinreal opened this issue · comments
MagicSource commented
I think multi class can not using BCELoss here, why are u using that?
Po-Chih Huang commented
The output and the target are both construct with 0 and 1 value.
So it's trainable with BCELoss.