test_epoch()

Run one evaluation pass and return true and predicted labels.

Usage

Source

test_epoch(
    model,
    validation_loader,
    device,
)

Parameters

model: nn.Module

Network model (must already be on device).

validation_loader: DataLoader[tuple[torch.Tensor, torch.Tensor]]

DataLoader yielding (images, labels) batches.

device: torch.device

Device to run inference on.

Returns

tuple[torch.Tensor, torch.Tensor]

Tuple of (y_true, y_pred) as 1-D CPU tensors.