Open HZD0014 opened 2 years ago
As far as I know, BCE is customarily used for binary classification problems. Wouldn't it be better to use CrossEntropyLoss
As far as I know, BCE is customarily used for binary classification problems. Wouldn't it be better to use CrossEntropyLoss