{
  "base_hidden_size": 3584,
  "sketch_dim": 256,
  "max_steps": 24,
  "step_max_len": 48,
  "label_keys": [
    "variable_aliasing",
    "unit_mismatch",
    "independence_assumption",
    "logic_wo_equations",
    "step_shuffle"
  ]
}