gr986 opened this issue 4 years ago · 0 comments
如果我想要把最后的结果改成二分类问题， 1、改动哪里的loss 2、改动哪里的输出 使模型输出一个[batch_size,2]大小的的prediction，和target做交叉熵损失？