nn_config.json

{
  "input_dim": 64,
  "hidden_dims": [
    512,
    256,
    128
  ],
  "dropout": 0.15
}