elektronn3.training.trainer_gnn_minibatch moduleΒΆ