- class ignite.metrics.Loss(loss_fn, output_transform=<function Loss.<lambda>>, batch_size=<built-in function len>, device=device(type='cpu'))#
Calculates the average loss according to the passed loss_fn.
loss_fn (Callable) – a callable taking a prediction tensor, a target tensor, optionally other arguments, and returns the average loss over all observations in the batch.
output_transform (Callable) – a callable that is used to transform the
process_function’s output into the form expected by the metric. This can be useful if, for example, you have a multi-output model and you want to compute the metric with respect to one of the outputs. The output is expected to be a tuple (prediction, target) or (prediction, target, kwargs) where kwargs is a dictionary of extra keywords arguments. If extra keywords arguments are provided they are passed to loss_fn.
batch_size (Callable) – a callable taking a target tensor that returns the first dimension size (usually the batch size).
device (Union[str, torch.device]) – specifies which device updates are accumulated on. Setting the metric’s device to be the same as your
updatearguments ensures the
updatemethod is non-blocking. By default, CPU.
dictionary defines required keys to be found in
engine.state.outputif the latter is a dictionary. Default,
("y_pred", "y", "criterion_kwargs"). This is useful when the criterion function requires additional arguments, which can be passed using
criterion_kwargs. See an example below.
Let’s implement a Loss metric that requires
criterion_kwargsas input for
criterionfunction. In the example below we show how to setup standard metric like Accuracy and the Loss metric using an
model = default_model criterion = nn.NLLLoss() metric = Loss(criterion) metric.attach(default_evaluator, 'loss') y_pred = torch.tensor([[0.1, 0.4, 0.5], [0.1, 0.7, 0.2]]) y_true = torch.tensor([2, 2]).long() state = default_evaluator.run([[y_pred, y_true]]) print(state.metrics['loss'])
Computes the metric based on it's accumulated state.
Resets the metric to it's initial state.
Updates the metric's state using the passed batch output.
Computes the metric based on it’s accumulated state.
By default, this is called at the end of each epoch.
- Return type
NotComputableError – raised when the metric cannot be computed.
Resets the metric to it’s initial state.
By default, this is called at the start of each epoch.
- Return type