oneflow.losses

Operators for neural networks

oneflow.losses.add_loss(loss: oneflow_api.BlobDesc) → None

Mark a Blob as a loss. Auto grad starts at every loss blob. It doesn’t has to be a product of typical “loss” operator like softmax loss but can also be a Blob produced by any operator.

Parameters

loss – A Blob.