analogvnn.fn.test#

Module Contents#

Functions#

test(→ Tuple[float, float])

Test the model on the test set.
analogvnn.fn.test.test(model: torch.nn.Module, test_loader: torch.utils.data.DataLoader, test_run: bool = False) → Tuple[float, float][source]#

Test the model on the test set.

Parameters:

  • model (torch.nn.Module) – the model to test.

  • test_loader (DataLoader) – the test set.

  • test_run (bool) – is it a test run.

Returns:

the loss and accuracy of the model on the test set.

Return type:

tuple