nvflare.app_opt.p2p.utils.metrics module

compute_loss_over_dataset(model: Module | None = None, loss: _Loss | None = None, dataloader: DataLoader | None = None, device: device | None = None) → float[source]

Compute the average loss over a dataset.

Parameters:

  • model – The model to use for predictions.

  • loss – The loss function to use.

  • dataloader – The dataloader for the dataset.

  • device – The device to use for computation.

Returns:

The average loss over the dataset.