graph_model: hidden_dim: 64 num_layers: 2 dropout: 0.1 batch_size: 32 epochs: 5