config.json

{
  "input_dim": 6,
  "hidden_dim": 512,
  "output_dim": 3,
  "n_layers": 3
}