pytorch中交叉熵的问题
2024-08-30
1 min read
F.cross_entropy() 是自带softmax的，因此不能在sigmoid之后直接送到该损失中。
binary_cross_entropy_with_logits() 为logits值，未经过sigmoid转换。
binary_cross_entropy() 和BCEloss需要输送的是概率值。