test.json

[
    {
        "type": "layer",
        "torch_module": "nn.Conv2d",
        "in_channels": 1,
        "out_channels": 6,
        "kernel_size": [2, 2],
        "stride": [1, 1],
        "padding": [0, 0]
    },
    {
        "type": "layer",
        "torch_module": "nn.ReLU"
    },
    {
        "type": "layer",
        "torch_module": "nn.MaxPool2d",
        "kernel_size": [2, 2],
        "stride": [2, 2]
    },
    {
        "type": "layer",
        "torch_module": "nn.Conv2d",
        "in_channels": 6,
        "out_channels": 16,
        "kernel_size": [5, 5],
        "stride": [1, 1],
        "padding": [0, 0]
    },
    {
        "type": "layer",
        "torch_module": "nn.ReLU"
    },
    {
        "type": "layer",
        "torch_module": "nn.MaxPool2d",
        "kernel_size": [2, 2],
        "stride": [2, 2]
    }
]