def forward_batch(model, batch, device):
    """Run a batch of data through a network/model."""
    inputs, labels = (batch, batch)
    inputs, labels = inputs.to(device), labels.to(device)

    outputs = model(inputs, labels=labels)

    return outputs[:2]