Bering.training.TrainerEdge.validate

TrainerEdge.validate(loader, image)[source]

Validate the model on the test set.

Parameters:

  • loader – Testing data loader: torch_geometric.data.DataLoader

  • image (Tensor) – image as the input for the image encoder