config.json

{
    "model_type": "Net",
    "num_classes": 4,
    "input_channels": 3,
    "input_size": [
        256,
        256
    ],
    "hidden_dim": 32,
    "activation": "tanh",
    "pooling": "max_pool2d"
}