config.json

{
  "input_dim": 6,
  "hidden_dim": 256,
  "output_dim": 3
}