# Baseline GIN model config
name: baseline
architecture: gin

encoder:
  num_layers: 3
  hidden_dim: 128
  dropout: 0.1

node_features_dim: 22
edge_features_dim: 22

heads:
  edge_classification:
    enabled: true
    hidden_dim: 64
  graph_classification:
    enabled: true
    num_classes: 4  # rigid, under, over, mixed
  joint_type:
    enabled: true
    num_classes: 12
  dof_regression:
    enabled: true
