config.json

{"input_dim": 784, "hidden_dims": [256, 64, 16, 4, 2]}