Bering.training.TrainerEdge.update

TrainerEdge.update(loader, image)[source]

Update the model on the training set.

Parameters:

loader – Training data loader: torch_geometric.data.DataLoader
image (Tensor) – image as the input for the image encoder