当num_classes=2时（做二分类问题时），出现错误 #18

Open

opened

on Jul 17, 2023

preds_softmax = preds_softmax.gather(1,labels.view(-1,1)) 处出现
index 2 is out of bounds for dimension 1 with size 2

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests