input val >=zero && input_val <=one

pytorch 训练时,计算loss时报错:

input val >=zero && input_val <=one

 

注意:

如果是BCEloss,需要加一个 sigmoid 或 sofmax 函数。

 

其他层:

网上的解决方法:在每一层cov,使用batch_norm一般可以解决问题

 

我遇到的情况:

focalloss时报的错

 

解决方法:

bceloss标签必须在0-1之间,

打印监控标签,看看什么原因:

   if targets_weighted.max()>1 or  targets_weighted.min()<0:
            print("label error",targets_weighted.max(),targets_weighted.min())

 

你可能感兴趣的:(pytorch知识宝典)