实例解释BCELoss与BCEWithLogitsLoss的关联(二分类问题)

发布时间 2023-04-25 14:20:32作者: BooTurbo

 

 

BCEWithLogitsLoss = Sigmoid + BCELoss,

 

 

 

 

 

 

nn接口                                              Function接口

nn.BCELoss( )                                  F.binary_cross_entropy( )

nn.BCEWithLogitsLoss( )                 F.binary_cross_entropy_with_logits( )