{ "stem.conv1.weight": { "shape": [ 40, 1, 3, 3, 3 ], "from_onnx": [ { "stack": [ "stem.conv1.convs.0.weight", "stem.conv1.convs.1.weight", "stem.conv1.convs.2.weight" ], "axis": 2 } ] }, "stem.conv1.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.conv1.convs.0.bias" } ] }, "stem.norm1.weight": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.norm1.weight" } ] }, "stem.norm1.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.norm1.bias" } ] }, "stem.conv2.weight": { "shape": [ 40, 40, 3, 3, 3 ], "from_onnx": [ { "stack": [ "stem.conv2.convs.0.weight", "stem.conv2.convs.1.weight", "stem.conv2.convs.2.weight" ], "axis": 2 } ] }, "stem.conv2.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.conv2.convs.0.bias" } ] }, "stem.norm2.weight": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.norm2.weight" } ] }, "stem.norm2.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.norm2.bias" } ] }, "stem.skip.weight": { "shape": [ 40, 1, 1, 1, 1 ], "from_onnx": [ { "stack": [ "stem.skip.convs.0.weight" ], "axis": 2 } ] }, "stem.skip.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "stem.skip.convs.0.bias" } ] }, "downs.0.conv.weight": { "shape": [ 80, 40, 2, 2, 2 ], "from_onnx": [ { "stack": [ "downs.0.conv.convs.0.weight", "downs.0.conv.convs.1.weight" ], "axis": 2 } ] }, "downs.0.conv.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "downs.0.conv.convs.0.bias" } ] }, "downs.1.conv.weight": { "shape": [ 160, 80, 2, 2, 2 ], "from_onnx": [ { "stack": [ "downs.1.conv.convs.0.weight", "downs.1.conv.convs.1.weight" ], "axis": 2 } ] }, "downs.1.conv.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "downs.1.conv.convs.0.bias" } ] }, "downs.2.conv.weight": { "shape": [ 320, 160, 2, 2, 2 ], "from_onnx": [ { "stack": [ "downs.2.conv.convs.0.weight", "downs.2.conv.convs.1.weight" ], "axis": 2 } ] }, "downs.2.conv.bias": { "shape": [ 320 ], "from_onnx": [ { "from": "downs.2.conv.convs.0.bias" } ] }, "enc_blocks.0.0.conv1.weight": { "shape": [ 80, 80, 3, 3, 3 ], "from_onnx": [ { "stack": [ "enc_blocks.0.0.conv1.convs.0.weight", "enc_blocks.0.0.conv1.convs.1.weight", "enc_blocks.0.0.conv1.convs.2.weight" ], "axis": 2 } ] }, "enc_blocks.0.0.conv1.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "enc_blocks.0.0.conv1.convs.0.bias" } ] }, "enc_blocks.0.0.norm1.weight": { "shape": [ 80 ], "from_onnx": [ { "from": "enc_blocks.0.0.norm1.weight" } ] }, "enc_blocks.0.0.norm1.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "enc_blocks.0.0.norm1.bias" } ] }, "enc_blocks.0.0.conv2.weight": { "shape": [ 80, 80, 3, 3, 3 ], "from_onnx": [ { "stack": [ "enc_blocks.0.0.conv2.convs.0.weight", "enc_blocks.0.0.conv2.convs.1.weight", "enc_blocks.0.0.conv2.convs.2.weight" ], "axis": 2 } ] }, "enc_blocks.0.0.conv2.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "enc_blocks.0.0.conv2.convs.0.bias" } ] }, "enc_blocks.0.0.norm2.weight": { "shape": [ 80 ], "from_onnx": [ { "from": "enc_blocks.0.0.norm2.weight" } ] }, "enc_blocks.0.0.norm2.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "enc_blocks.0.0.norm2.bias" } ] }, "enc_blocks.1.0.conv1.weight": { "shape": [ 160, 160, 3, 3, 3 ], "from_onnx": [ { "stack": [ "enc_blocks.1.0.conv1.convs.0.weight", "enc_blocks.1.0.conv1.convs.1.weight", "enc_blocks.1.0.conv1.convs.2.weight" ], "axis": 2 } ] }, "enc_blocks.1.0.conv1.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "enc_blocks.1.0.conv1.convs.0.bias" } ] }, "enc_blocks.1.0.norm1.weight": { "shape": [ 160 ], "from_onnx": [ { "from": "enc_blocks.1.0.norm1.weight" } ] }, "enc_blocks.1.0.norm1.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "enc_blocks.1.0.norm1.bias" } ] }, "enc_blocks.1.0.conv2.weight": { "shape": [ 160, 160, 3, 3, 3 ], "from_onnx": [ { "stack": [ "enc_blocks.1.0.conv2.convs.0.weight", "enc_blocks.1.0.conv2.convs.1.weight", "enc_blocks.1.0.conv2.convs.2.weight" ], "axis": 2 } ] }, "enc_blocks.1.0.conv2.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "enc_blocks.1.0.conv2.convs.0.bias" } ] }, "enc_blocks.1.0.norm2.weight": { "shape": [ 160 ], "from_onnx": [ { "from": "enc_blocks.1.0.norm2.weight" } ] }, "enc_blocks.1.0.norm2.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "enc_blocks.1.0.norm2.bias" } ] }, "enc_blocks.2.0.conv1.weight": { "shape": [ 320, 320, 3, 3, 3 ], "from_onnx": [ { "stack": [ "enc_blocks.2.0.conv1.convs.0.weight", "enc_blocks.2.0.conv1.convs.1.weight", "enc_blocks.2.0.conv1.convs.2.weight" ], "axis": 2 } ] }, "enc_blocks.2.0.conv1.bias": { "shape": [ 320 ], "from_onnx": [ { "from": "enc_blocks.2.0.conv1.convs.0.bias" } ] }, "enc_blocks.2.0.norm1.weight": { "shape": [ 320 ], "from_onnx": [ { "from": "enc_blocks.2.0.norm1.weight" } ] }, "enc_blocks.2.0.norm1.bias": { "shape": [ 320 ], "from_onnx": [ { "from": "enc_blocks.2.0.norm1.bias" } ] }, "enc_blocks.2.0.conv2.weight": { "shape": [ 320, 320, 3, 3, 3 ], "from_onnx": [ { "stack": [ "enc_blocks.2.0.conv2.convs.0.weight", "enc_blocks.2.0.conv2.convs.1.weight", "enc_blocks.2.0.conv2.convs.2.weight" ], "axis": 2 } ] }, "enc_blocks.2.0.conv2.bias": { "shape": [ 320 ], "from_onnx": [ { "from": "enc_blocks.2.0.conv2.convs.0.bias" } ] }, "enc_blocks.2.0.norm2.weight": { "shape": [ 320 ], "from_onnx": [ { "from": "enc_blocks.2.0.norm2.weight" } ] }, "enc_blocks.2.0.norm2.bias": { "shape": [ 320 ], "from_onnx": [ { "from": "enc_blocks.2.0.norm2.bias" } ] }, "ups.0.reduce.weight": { "shape": [ 160, 320, 1, 1, 1 ], "from_onnx": [ { "stack": [ "ups.0.reduce.convs.0.weight" ], "axis": 2 } ] }, "ups.0.reduce.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "ups.0.reduce.convs.0.bias" } ] }, "ups.1.reduce.weight": { "shape": [ 80, 160, 1, 1, 1 ], "from_onnx": [ { "stack": [ "ups.1.reduce.convs.0.weight" ], "axis": 2 } ] }, "ups.1.reduce.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "ups.1.reduce.convs.0.bias" } ] }, "ups.2.reduce.weight": { "shape": [ 40, 80, 1, 1, 1 ], "from_onnx": [ { "stack": [ "ups.2.reduce.convs.0.weight" ], "axis": 2 } ] }, "ups.2.reduce.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "ups.2.reduce.convs.0.bias" } ] }, "dec_blocks.0.0.conv1.weight": { "shape": [ 160, 320, 3, 3, 3 ], "from_onnx": [ { "stack": [ "dec_blocks.0.0.conv1.convs.0.weight", "dec_blocks.0.0.conv1.convs.1.weight", "dec_blocks.0.0.conv1.convs.2.weight" ], "axis": 2 } ] }, "dec_blocks.0.0.conv1.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.conv1.convs.0.bias" } ] }, "dec_blocks.0.0.norm1.weight": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.norm1.weight" } ] }, "dec_blocks.0.0.norm1.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.norm1.bias" } ] }, "dec_blocks.0.0.conv2.weight": { "shape": [ 160, 160, 3, 3, 3 ], "from_onnx": [ { "stack": [ "dec_blocks.0.0.conv2.convs.0.weight", "dec_blocks.0.0.conv2.convs.1.weight", "dec_blocks.0.0.conv2.convs.2.weight" ], "axis": 2 } ] }, "dec_blocks.0.0.conv2.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.conv2.convs.0.bias" } ] }, "dec_blocks.0.0.norm2.weight": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.norm2.weight" } ] }, "dec_blocks.0.0.norm2.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.norm2.bias" } ] }, "dec_blocks.0.0.skip.weight": { "shape": [ 160, 320, 1, 1, 1 ], "from_onnx": [ { "stack": [ "dec_blocks.0.0.skip.convs.0.weight" ], "axis": 2 } ] }, "dec_blocks.0.0.skip.bias": { "shape": [ 160 ], "from_onnx": [ { "from": "dec_blocks.0.0.skip.convs.0.bias" } ] }, "dec_blocks.1.0.conv1.weight": { "shape": [ 80, 160, 3, 3, 3 ], "from_onnx": [ { "stack": [ "dec_blocks.1.0.conv1.convs.0.weight", "dec_blocks.1.0.conv1.convs.1.weight", "dec_blocks.1.0.conv1.convs.2.weight" ], "axis": 2 } ] }, "dec_blocks.1.0.conv1.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.conv1.convs.0.bias" } ] }, "dec_blocks.1.0.norm1.weight": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.norm1.weight" } ] }, "dec_blocks.1.0.norm1.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.norm1.bias" } ] }, "dec_blocks.1.0.conv2.weight": { "shape": [ 80, 80, 3, 3, 3 ], "from_onnx": [ { "stack": [ "dec_blocks.1.0.conv2.convs.0.weight", "dec_blocks.1.0.conv2.convs.1.weight", "dec_blocks.1.0.conv2.convs.2.weight" ], "axis": 2 } ] }, "dec_blocks.1.0.conv2.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.conv2.convs.0.bias" } ] }, "dec_blocks.1.0.norm2.weight": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.norm2.weight" } ] }, "dec_blocks.1.0.norm2.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.norm2.bias" } ] }, "dec_blocks.1.0.skip.weight": { "shape": [ 80, 160, 1, 1, 1 ], "from_onnx": [ { "stack": [ "dec_blocks.1.0.skip.convs.0.weight" ], "axis": 2 } ] }, "dec_blocks.1.0.skip.bias": { "shape": [ 80 ], "from_onnx": [ { "from": "dec_blocks.1.0.skip.convs.0.bias" } ] }, "dec_blocks.2.0.conv1.weight": { "shape": [ 40, 80, 3, 3, 3 ], "from_onnx": [ { "stack": [ "dec_blocks.2.0.conv1.convs.0.weight", "dec_blocks.2.0.conv1.convs.1.weight", "dec_blocks.2.0.conv1.convs.2.weight" ], "axis": 2 } ] }, "dec_blocks.2.0.conv1.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.conv1.convs.0.bias" } ] }, "dec_blocks.2.0.norm1.weight": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.norm1.weight" } ] }, "dec_blocks.2.0.norm1.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.norm1.bias" } ] }, "dec_blocks.2.0.conv2.weight": { "shape": [ 40, 40, 3, 3, 3 ], "from_onnx": [ { "stack": [ "dec_blocks.2.0.conv2.convs.0.weight", "dec_blocks.2.0.conv2.convs.1.weight", "dec_blocks.2.0.conv2.convs.2.weight" ], "axis": 2 } ] }, "dec_blocks.2.0.conv2.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.conv2.convs.0.bias" } ] }, "dec_blocks.2.0.norm2.weight": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.norm2.weight" } ] }, "dec_blocks.2.0.norm2.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.norm2.bias" } ] }, "dec_blocks.2.0.skip.weight": { "shape": [ 40, 80, 1, 1, 1 ], "from_onnx": [ { "stack": [ "dec_blocks.2.0.skip.convs.0.weight" ], "axis": 2 } ] }, "dec_blocks.2.0.skip.bias": { "shape": [ 40 ], "from_onnx": [ { "from": "dec_blocks.2.0.skip.convs.0.bias" } ] }, "final_conv.weight": { "shape": [ 1, 40, 1, 1, 1 ], "from_onnx": [ { "stack": [ "final_conv.convs.0.weight" ], "axis": 2 } ] }, "final_conv.bias": { "shape": [ 1 ], "from_onnx": [ { "from": "final_conv.convs.0.bias" } ] } }