Python tf.nn.compute_average_loss用法及代码示例

用sample_weights 缩放per-example 损失并计算它们的平均值。

用法

tf.nn.compute_average_loss(
    per_example_loss, sample_weight=None, global_batch_size=None
)

与分配策略和自定义训练循环一起使用：

with strategy.scope():
  def compute_loss(labels, predictions, sample_weight=None):

    # If you are using a `Loss` class instead, set reduction to `NONE` so that
    # we can do the reduction afterwards and divide by global batch size.
    per_example_loss = tf.keras.losses.sparse_categorical_crossentropy(
        labels, predictions)

    # Compute loss that is scaled by sample_weight and by global batch size.
    return tf.nn.compute_average_loss(
        per_example_loss,
        sample_weight=sample_weight,
        global_batch_size=GLOBAL_BATCH_SIZE)