33fde6b3013694aa9721bb24996bb6ff28a04d70,returnn/tf/layers/basic.py,BinaryCrossEntropyLoss,get_value,#BinaryCrossEntropyLoss#,6653

Before Change

    :rtype: tf.Tensor
    
    with tf.name_scope("loss_bin_ce"):
      out = tf.nn.sigmoid_cross_entropy_with_logits(
        logits=self.output_flat, labels=tf.cast(self.target_flat, self.output_flat.dtype))
      return self.reduce_func(out) * (1.0 / (self.output.dim or 1))

  def get_error(self):
    

After Change

    with tf.name_scope("loss_bin_ce"):
      target_flat = tf.cast(self.target_flat, self.output_flat.dtype)
      if self._pos_weight is None:
        out = tf.nn.sigmoid_cross_entropy_with_logits(logits=self.output_flat, labels=target_flat)
      else:
        out = tf.nn.weighted_cross_entropy_with_logits(
          logits=self.output_flat, labels=target_flat, pos_weight=self._pos_weight)
      return self.reduce_func(out) * (1.0 / (self.output.dim or 1))

Instances