Liam
Liam
> 我跑了一下，loss 一直不收敛，不知道是哪里出问题了 我的也是，而且loss还是负数。。。
> 我觉得核心的问题所在是这个F.binary_cross_entropy_with_logits()应该加上reduction='none'参数。否则计算的结果会求平均值，输出loss只有一个值，无法和后面的那些object_mask做element-wise的乘法. F.binary_cross_entropy_with_logits(predictions,targets) -> F.binary_cross_entropy_with_logits(predictions,targets,reduction='none') 请问您能分享一下吗 魔改后的