Bering.training.TrainerNode.update

TrainerNode.update(loader)[source]

Update the model on the training set.

Parameters:: loader – Training data loader: torch_geometric.data.DataLoader