{
  "encoder.conv1.weight": {
    "shape": [
      160,
      12,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 207360
  },
  "encoder.conv1.bias": {
    "shape": [
      160
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.0.residual.0.gamma": {
    "shape": [
      160,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.0.residual.2.weight": {
    "shape": [
      160,
      160,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 2764800
  },
  "encoder.downsamples.0.downsamples.0.residual.2.bias": {
    "shape": [
      160
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.0.residual.3.gamma": {
    "shape": [
      160,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.0.residual.6.weight": {
    "shape": [
      160,
      160,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 2764800
  },
  "encoder.downsamples.0.downsamples.0.residual.6.bias": {
    "shape": [
      160
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.1.residual.0.gamma": {
    "shape": [
      160,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.1.residual.2.weight": {
    "shape": [
      160,
      160,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 2764800
  },
  "encoder.downsamples.0.downsamples.1.residual.2.bias": {
    "shape": [
      160
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.1.residual.3.gamma": {
    "shape": [
      160,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.1.residual.6.weight": {
    "shape": [
      160,
      160,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 2764800
  },
  "encoder.downsamples.0.downsamples.1.residual.6.bias": {
    "shape": [
      160
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.0.downsamples.2.resample.1.weight": {
    "shape": [
      160,
      160,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 921600
  },
  "encoder.downsamples.0.downsamples.2.resample.1.bias": {
    "shape": [
      160
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.1.downsamples.0.residual.0.gamma": {
    "shape": [
      160,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 640
  },
  "encoder.downsamples.1.downsamples.0.residual.2.weight": {
    "shape": [
      320,
      160,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 5529600
  },
  "encoder.downsamples.1.downsamples.0.residual.2.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.0.residual.3.gamma": {
    "shape": [
      320,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.0.residual.6.weight": {
    "shape": [
      320,
      320,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 11059200
  },
  "encoder.downsamples.1.downsamples.0.residual.6.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.0.shortcut.weight": {
    "shape": [
      320,
      160,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 204800
  },
  "encoder.downsamples.1.downsamples.0.shortcut.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.1.residual.0.gamma": {
    "shape": [
      320,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.1.residual.2.weight": {
    "shape": [
      320,
      320,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 11059200
  },
  "encoder.downsamples.1.downsamples.1.residual.2.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.1.residual.3.gamma": {
    "shape": [
      320,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.1.residual.6.weight": {
    "shape": [
      320,
      320,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 11059200
  },
  "encoder.downsamples.1.downsamples.1.residual.6.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.2.resample.1.weight": {
    "shape": [
      320,
      320,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 3686400
  },
  "encoder.downsamples.1.downsamples.2.resample.1.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.1.downsamples.2.time_conv.weight": {
    "shape": [
      320,
      320,
      3,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1228800
  },
  "encoder.downsamples.1.downsamples.2.time_conv.bias": {
    "shape": [
      320
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.2.downsamples.0.residual.0.gamma": {
    "shape": [
      320,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1280
  },
  "encoder.downsamples.2.downsamples.0.residual.2.weight": {
    "shape": [
      640,
      320,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 22118400
  },
  "encoder.downsamples.2.downsamples.0.residual.2.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.0.residual.3.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.0.residual.6.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.2.downsamples.0.residual.6.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.0.shortcut.weight": {
    "shape": [
      640,
      320,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 819200
  },
  "encoder.downsamples.2.downsamples.0.shortcut.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.1.residual.0.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.1.residual.2.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.2.downsamples.1.residual.2.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.1.residual.3.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.1.residual.6.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.2.downsamples.1.residual.6.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.2.resample.1.weight": {
    "shape": [
      640,
      640,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 14745600
  },
  "encoder.downsamples.2.downsamples.2.resample.1.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.2.downsamples.2.time_conv.weight": {
    "shape": [
      640,
      640,
      3,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4915200
  },
  "encoder.downsamples.2.downsamples.2.time_conv.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.0.residual.0.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.0.residual.2.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.3.downsamples.0.residual.2.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.0.residual.3.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.0.residual.6.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.3.downsamples.0.residual.6.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.1.residual.0.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.1.residual.2.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.3.downsamples.1.residual.2.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.1.residual.3.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.downsamples.3.downsamples.1.residual.6.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.downsamples.3.downsamples.1.residual.6.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.0.residual.0.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.0.residual.2.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.middle.0.residual.2.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.0.residual.3.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.0.residual.6.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.middle.0.residual.6.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.1.norm.gamma": {
    "shape": [
      640,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.1.to_qkv.weight": {
    "shape": [
      1920,
      640,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4915200
  },
  "encoder.middle.1.to_qkv.bias": {
    "shape": [
      1920
    ],
    "dtype": "torch.float32",
    "bytes": 7680
  },
  "encoder.middle.1.proj.weight": {
    "shape": [
      640,
      640,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1638400
  },
  "encoder.middle.1.proj.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.2.residual.0.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.2.residual.2.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.middle.2.residual.2.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.2.residual.3.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.middle.2.residual.6.weight": {
    "shape": [
      640,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 44236800
  },
  "encoder.middle.2.residual.6.bias": {
    "shape": [
      640
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.head.0.gamma": {
    "shape": [
      640,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2560
  },
  "encoder.head.2.weight": {
    "shape": [
      96,
      640,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 6635520
  },
  "encoder.head.2.bias": {
    "shape": [
      96
    ],
    "dtype": "torch.float32",
    "bytes": 384
  },
  "conv1.weight": {
    "shape": [
      96,
      96,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 36864
  },
  "conv1.bias": {
    "shape": [
      96
    ],
    "dtype": "torch.float32",
    "bytes": 384
  },
  "conv2.weight": {
    "shape": [
      48,
      48,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 9216
  },
  "conv2.bias": {
    "shape": [
      48
    ],
    "dtype": "torch.float32",
    "bytes": 192
  },
  "decoder.conv1.weight": {
    "shape": [
      1024,
      48,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 5308416
  },
  "decoder.conv1.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.0.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.0.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.middle.0.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.0.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.0.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.middle.0.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.1.norm.gamma": {
    "shape": [
      1024,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.1.to_qkv.weight": {
    "shape": [
      3072,
      1024,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 12582912
  },
  "decoder.middle.1.to_qkv.bias": {
    "shape": [
      3072
    ],
    "dtype": "torch.float32",
    "bytes": 12288
  },
  "decoder.middle.1.proj.weight": {
    "shape": [
      1024,
      1024,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4194304
  },
  "decoder.middle.1.proj.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.2.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.2.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.middle.2.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.2.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.middle.2.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.middle.2.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.0.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.0.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.0.upsamples.0.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.0.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.0.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.0.upsamples.0.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.1.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.1.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.0.upsamples.1.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.1.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.1.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.0.upsamples.1.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.2.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.2.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.0.upsamples.2.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.2.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.2.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.0.upsamples.2.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.3.resample.1.weight": {
    "shape": [
      1024,
      1024,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 37748736
  },
  "decoder.upsamples.0.upsamples.3.resample.1.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.0.upsamples.3.time_conv.weight": {
    "shape": [
      2048,
      1024,
      3,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 25165824
  },
  "decoder.upsamples.0.upsamples.3.time_conv.bias": {
    "shape": [
      2048
    ],
    "dtype": "torch.float32",
    "bytes": 8192
  },
  "decoder.upsamples.1.upsamples.0.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.0.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.1.upsamples.0.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.0.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.0.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.1.upsamples.0.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.1.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.1.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.1.upsamples.1.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.1.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.1.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.1.upsamples.1.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.2.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.2.residual.2.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.1.upsamples.2.residual.2.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.2.residual.3.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.2.residual.6.weight": {
    "shape": [
      1024,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 113246208
  },
  "decoder.upsamples.1.upsamples.2.residual.6.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.3.resample.1.weight": {
    "shape": [
      1024,
      1024,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 37748736
  },
  "decoder.upsamples.1.upsamples.3.resample.1.bias": {
    "shape": [
      1024
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.1.upsamples.3.time_conv.weight": {
    "shape": [
      2048,
      1024,
      3,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 25165824
  },
  "decoder.upsamples.1.upsamples.3.time_conv.bias": {
    "shape": [
      2048
    ],
    "dtype": "torch.float32",
    "bytes": 8192
  },
  "decoder.upsamples.2.upsamples.0.residual.0.gamma": {
    "shape": [
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 4096
  },
  "decoder.upsamples.2.upsamples.0.residual.2.weight": {
    "shape": [
      512,
      1024,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 56623104
  },
  "decoder.upsamples.2.upsamples.0.residual.2.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.0.residual.3.gamma": {
    "shape": [
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.0.residual.6.weight": {
    "shape": [
      512,
      512,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 28311552
  },
  "decoder.upsamples.2.upsamples.0.residual.6.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.0.shortcut.weight": {
    "shape": [
      512,
      1024,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2097152
  },
  "decoder.upsamples.2.upsamples.0.shortcut.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.1.residual.0.gamma": {
    "shape": [
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.1.residual.2.weight": {
    "shape": [
      512,
      512,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 28311552
  },
  "decoder.upsamples.2.upsamples.1.residual.2.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.1.residual.3.gamma": {
    "shape": [
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.1.residual.6.weight": {
    "shape": [
      512,
      512,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 28311552
  },
  "decoder.upsamples.2.upsamples.1.residual.6.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.2.residual.0.gamma": {
    "shape": [
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.2.residual.2.weight": {
    "shape": [
      512,
      512,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 28311552
  },
  "decoder.upsamples.2.upsamples.2.residual.2.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.2.residual.3.gamma": {
    "shape": [
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.2.residual.6.weight": {
    "shape": [
      512,
      512,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 28311552
  },
  "decoder.upsamples.2.upsamples.2.residual.6.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.2.upsamples.3.resample.1.weight": {
    "shape": [
      512,
      512,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 9437184
  },
  "decoder.upsamples.2.upsamples.3.resample.1.bias": {
    "shape": [
      512
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.3.upsamples.0.residual.0.gamma": {
    "shape": [
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 2048
  },
  "decoder.upsamples.3.upsamples.0.residual.2.weight": {
    "shape": [
      256,
      512,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 14155776
  },
  "decoder.upsamples.3.upsamples.0.residual.2.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.0.residual.3.gamma": {
    "shape": [
      256,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.0.residual.6.weight": {
    "shape": [
      256,
      256,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 7077888
  },
  "decoder.upsamples.3.upsamples.0.residual.6.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.0.shortcut.weight": {
    "shape": [
      256,
      512,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 524288
  },
  "decoder.upsamples.3.upsamples.0.shortcut.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.1.residual.0.gamma": {
    "shape": [
      256,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.1.residual.2.weight": {
    "shape": [
      256,
      256,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 7077888
  },
  "decoder.upsamples.3.upsamples.1.residual.2.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.1.residual.3.gamma": {
    "shape": [
      256,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.1.residual.6.weight": {
    "shape": [
      256,
      256,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 7077888
  },
  "decoder.upsamples.3.upsamples.1.residual.6.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.2.residual.0.gamma": {
    "shape": [
      256,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.2.residual.2.weight": {
    "shape": [
      256,
      256,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 7077888
  },
  "decoder.upsamples.3.upsamples.2.residual.2.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.2.residual.3.gamma": {
    "shape": [
      256,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.upsamples.3.upsamples.2.residual.6.weight": {
    "shape": [
      256,
      256,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 7077888
  },
  "decoder.upsamples.3.upsamples.2.residual.6.bias": {
    "shape": [
      256
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.head.0.gamma": {
    "shape": [
      256,
      1,
      1,
      1
    ],
    "dtype": "torch.float32",
    "bytes": 1024
  },
  "decoder.head.2.weight": {
    "shape": [
      12,
      256,
      3,
      3,
      3
    ],
    "dtype": "torch.float32",
    "bytes": 331776
  },
  "decoder.head.2.bias": {
    "shape": [
      12
    ],
    "dtype": "torch.float32",
    "bytes": 48
  }
}