config.json

{"model_type": "custom", "hidden_size": 256, "num_layers": 3, "dropout": 0.2}