Loading

2022年12月29日
pytorch：二分类时的loss选择
摘要： PyTorch二分类时BCELoss，CrossEntropyLoss，Sigmoid等的选择和使用 这里就总结一下使用PyTorch做二分类时的几种情况： 总体上来讲，有三种实现形式： 最后分类层降至一维，使用sigmoid输出一个0-1之间的分数，使用torch.nn.BCELoss作为loss 阅读全文
posted @ 2022-12-29 11:37 戴墨镜的长颈鹿 阅读(1550) 评论(0) 推荐(0) 编辑