model.json

{
  "header": "A model architecture for 256^3 brains with 5 decoders, where dilation in each is structured as 16->8->4->2->1",
  "bnorm": true,
  "gelu": true,
  "dropout_p": 0,
  "layers": [
    {
      "in_channels": -1,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 16,
      "stride": 1,
      "dilation": 16
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 8,
      "stride": 1,
      "dilation": 8
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 4,
      "stride": 1,
      "dilation": 4
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 2,
      "stride": 1,
      "dilation": 2
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 1,
      "stride": 1,
      "dilation": 1
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 16,
      "stride": 1,
      "dilation": 16
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 8,
      "stride": 1,
      "dilation": 8
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 4,
      "stride": 1,
      "dilation": 4
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 2,
      "stride": 1,
      "dilation": 2
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 1,
      "stride": 1,
      "dilation": 1
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 16,
      "stride": 1,
      "dilation": 16
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 8,
      "stride": 1,
      "dilation": 8
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 4,
      "stride": 1,
      "dilation": 4
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 2,
      "stride": 1,
      "dilation": 2
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 1,
      "stride": 1,
      "dilation": 1
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 16,
      "stride": 1,
      "dilation": 16
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 8,
      "stride": 1,
      "dilation": 8
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 4,
      "stride": 1,
      "dilation": 4
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 2,
      "stride": 1,
      "dilation": 2
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 1,
      "stride": 1,
      "dilation": 1
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 16,
      "stride": 1,
      "dilation": 16
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 8,
      "stride": 1,
      "dilation": 8
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 4,
      "stride": 1,
      "dilation": 4
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 2,
      "stride": 1,
      "dilation": 2
    },
    {
      "in_channels": 5,
      "out_channels": 5,
      "kernel_size": 3,
      "padding": 1,
      "stride": 1,
      "dilation": 1
    },
    {
      "in_channels": 5,
      "out_channels": -1,
      "kernel_size": 1,
      "padding": 0,
      "stride": 1,
      "dilation": 1
    }
  ]
}