# Advanced GAT model config
name: gat_solver
architecture: gat

encoder:
  num_layers: 4
  hidden_dim: 256
  num_heads: 8
  dropout: 0.1
  residual: true

node_features_dim: 22
edge_features_dim: 22

heads:
  edge_classification:
    enabled: true
    hidden_dim: 128
  graph_classification:
    enabled: true
    num_classes: 4
  joint_type:
    enabled: true
    num_classes: 12
  dof_regression:
    enabled: true
  dof_tracking:
    enabled: true
