{ "stem.conv1.convs.0.weight": { "torch": "stem.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "stem.conv1.convs.0.bias": { "torch": "stem.conv1.bias", "transform": "identity", "to_onnx": [] }, "stem.conv1.convs.1.weight": { "torch": "stem.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "stem.conv1.convs.2.weight": { "torch": "stem.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "stem.norm1.weight": { "torch": "stem.norm1.weight", "transform": "identity", "to_onnx": [] }, "stem.norm1.bias": { "torch": "stem.norm1.bias", "transform": "identity", "to_onnx": [] }, "stem.conv2.convs.0.weight": { "torch": "stem.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "stem.conv2.convs.0.bias": { "torch": "stem.conv2.bias", "transform": "identity", "to_onnx": [] }, "stem.conv2.convs.1.weight": { "torch": "stem.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "stem.conv2.convs.2.weight": { "torch": "stem.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "stem.norm2.weight": { "torch": "stem.norm2.weight", "transform": "identity", "to_onnx": [] }, "stem.norm2.bias": { "torch": "stem.norm2.bias", "transform": "identity", "to_onnx": [] }, "stem.skip.convs.0.weight": { "torch": "stem.skip.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "stem.skip.convs.0.bias": { "torch": "stem.skip.bias", "transform": "identity", "to_onnx": [] }, "downs.0.conv.convs.0.weight": { "torch": "downs.0.conv.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "downs.0.conv.convs.0.bias": { "torch": "downs.0.conv.bias", "transform": "identity", "to_onnx": [] }, "downs.0.conv.convs.1.weight": { "torch": "downs.0.conv.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "downs.1.conv.convs.0.weight": { "torch": "downs.1.conv.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "downs.1.conv.convs.0.bias": { "torch": "downs.1.conv.bias", "transform": "identity", "to_onnx": [] }, "downs.1.conv.convs.1.weight": { "torch": "downs.1.conv.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "downs.2.conv.convs.0.weight": { "torch": "downs.2.conv.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "downs.2.conv.convs.0.bias": { "torch": "downs.2.conv.bias", "transform": "identity", "to_onnx": [] }, "downs.2.conv.convs.1.weight": { "torch": "downs.2.conv.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.0.0.conv1.convs.0.weight": { "torch": "enc_blocks.0.0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "enc_blocks.0.0.conv1.convs.0.bias": { "torch": "enc_blocks.0.0.conv1.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.0.0.conv1.convs.1.weight": { "torch": "enc_blocks.0.0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.0.0.conv1.convs.2.weight": { "torch": "enc_blocks.0.0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "enc_blocks.0.0.norm1.weight": { "torch": "enc_blocks.0.0.norm1.weight", "transform": "identity", "to_onnx": [] }, "enc_blocks.0.0.norm1.bias": { "torch": "enc_blocks.0.0.norm1.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.0.0.conv2.convs.0.weight": { "torch": "enc_blocks.0.0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "enc_blocks.0.0.conv2.convs.0.bias": { "torch": "enc_blocks.0.0.conv2.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.0.0.conv2.convs.1.weight": { "torch": "enc_blocks.0.0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.0.0.conv2.convs.2.weight": { "torch": "enc_blocks.0.0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "enc_blocks.0.0.norm2.weight": { "torch": "enc_blocks.0.0.norm2.weight", "transform": "identity", "to_onnx": [] }, "enc_blocks.0.0.norm2.bias": { "torch": "enc_blocks.0.0.norm2.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.1.0.conv1.convs.0.weight": { "torch": "enc_blocks.1.0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "enc_blocks.1.0.conv1.convs.0.bias": { "torch": "enc_blocks.1.0.conv1.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.1.0.conv1.convs.1.weight": { "torch": "enc_blocks.1.0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.1.0.conv1.convs.2.weight": { "torch": "enc_blocks.1.0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "enc_blocks.1.0.norm1.weight": { "torch": "enc_blocks.1.0.norm1.weight", "transform": "identity", "to_onnx": [] }, "enc_blocks.1.0.norm1.bias": { "torch": "enc_blocks.1.0.norm1.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.1.0.conv2.convs.0.weight": { "torch": "enc_blocks.1.0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "enc_blocks.1.0.conv2.convs.0.bias": { "torch": "enc_blocks.1.0.conv2.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.1.0.conv2.convs.1.weight": { "torch": "enc_blocks.1.0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.1.0.conv2.convs.2.weight": { "torch": "enc_blocks.1.0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "enc_blocks.1.0.norm2.weight": { "torch": "enc_blocks.1.0.norm2.weight", "transform": "identity", "to_onnx": [] }, "enc_blocks.1.0.norm2.bias": { "torch": "enc_blocks.1.0.norm2.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.2.0.conv1.convs.0.weight": { "torch": "enc_blocks.2.0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "enc_blocks.2.0.conv1.convs.0.bias": { "torch": "enc_blocks.2.0.conv1.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.2.0.conv1.convs.1.weight": { "torch": "enc_blocks.2.0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.2.0.conv1.convs.2.weight": { "torch": "enc_blocks.2.0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "enc_blocks.2.0.norm1.weight": { "torch": "enc_blocks.2.0.norm1.weight", "transform": "identity", "to_onnx": [] }, "enc_blocks.2.0.norm1.bias": { "torch": "enc_blocks.2.0.norm1.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.2.0.conv2.convs.0.weight": { "torch": "enc_blocks.2.0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "enc_blocks.2.0.conv2.convs.0.bias": { "torch": "enc_blocks.2.0.conv2.bias", "transform": "identity", "to_onnx": [] }, "enc_blocks.2.0.conv2.convs.1.weight": { "torch": "enc_blocks.2.0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "enc_blocks.2.0.conv2.convs.2.weight": { "torch": "enc_blocks.2.0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "enc_blocks.2.0.norm2.weight": { "torch": "enc_blocks.2.0.norm2.weight", "transform": "identity", "to_onnx": [] }, "enc_blocks.2.0.norm2.bias": { "torch": "enc_blocks.2.0.norm2.bias", "transform": "identity", "to_onnx": [] }, "ups.0.reduce.convs.0.weight": { "torch": "ups.0.reduce.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "ups.0.reduce.convs.0.bias": { "torch": "ups.0.reduce.bias", "transform": "identity", "to_onnx": [] }, "ups.1.reduce.convs.0.weight": { "torch": "ups.1.reduce.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "ups.1.reduce.convs.0.bias": { "torch": "ups.1.reduce.bias", "transform": "identity", "to_onnx": [] }, "ups.2.reduce.convs.0.weight": { "torch": "ups.2.reduce.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "ups.2.reduce.convs.0.bias": { "torch": "ups.2.reduce.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.conv1.convs.0.weight": { "torch": "dec_blocks.0.0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.0.0.conv1.convs.0.bias": { "torch": "dec_blocks.0.0.conv1.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.conv1.convs.1.weight": { "torch": "dec_blocks.0.0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "dec_blocks.0.0.conv1.convs.2.weight": { "torch": "dec_blocks.0.0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "dec_blocks.0.0.norm1.weight": { "torch": "dec_blocks.0.0.norm1.weight", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.norm1.bias": { "torch": "dec_blocks.0.0.norm1.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.conv2.convs.0.weight": { "torch": "dec_blocks.0.0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.0.0.conv2.convs.0.bias": { "torch": "dec_blocks.0.0.conv2.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.conv2.convs.1.weight": { "torch": "dec_blocks.0.0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "dec_blocks.0.0.conv2.convs.2.weight": { "torch": "dec_blocks.0.0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "dec_blocks.0.0.norm2.weight": { "torch": "dec_blocks.0.0.norm2.weight", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.norm2.bias": { "torch": "dec_blocks.0.0.norm2.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.0.0.skip.convs.0.weight": { "torch": "dec_blocks.0.0.skip.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.0.0.skip.convs.0.bias": { "torch": "dec_blocks.0.0.skip.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.conv1.convs.0.weight": { "torch": "dec_blocks.1.0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.1.0.conv1.convs.0.bias": { "torch": "dec_blocks.1.0.conv1.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.conv1.convs.1.weight": { "torch": "dec_blocks.1.0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "dec_blocks.1.0.conv1.convs.2.weight": { "torch": "dec_blocks.1.0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "dec_blocks.1.0.norm1.weight": { "torch": "dec_blocks.1.0.norm1.weight", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.norm1.bias": { "torch": "dec_blocks.1.0.norm1.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.conv2.convs.0.weight": { "torch": "dec_blocks.1.0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.1.0.conv2.convs.0.bias": { "torch": "dec_blocks.1.0.conv2.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.conv2.convs.1.weight": { "torch": "dec_blocks.1.0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "dec_blocks.1.0.conv2.convs.2.weight": { "torch": "dec_blocks.1.0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "dec_blocks.1.0.norm2.weight": { "torch": "dec_blocks.1.0.norm2.weight", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.norm2.bias": { "torch": "dec_blocks.1.0.norm2.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.1.0.skip.convs.0.weight": { "torch": "dec_blocks.1.0.skip.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.1.0.skip.convs.0.bias": { "torch": "dec_blocks.1.0.skip.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.conv1.convs.0.weight": { "torch": "dec_blocks.2.0.conv1.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.2.0.conv1.convs.0.bias": { "torch": "dec_blocks.2.0.conv1.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.conv1.convs.1.weight": { "torch": "dec_blocks.2.0.conv1.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "dec_blocks.2.0.conv1.convs.2.weight": { "torch": "dec_blocks.2.0.conv1.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "dec_blocks.2.0.norm1.weight": { "torch": "dec_blocks.2.0.norm1.weight", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.norm1.bias": { "torch": "dec_blocks.2.0.norm1.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.conv2.convs.0.weight": { "torch": "dec_blocks.2.0.conv2.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.2.0.conv2.convs.0.bias": { "torch": "dec_blocks.2.0.conv2.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.conv2.convs.1.weight": { "torch": "dec_blocks.2.0.conv2.weight", "transform": "[:, :, 1]", "to_onnx": [ { "index": [ 2, 1 ] } ] }, "dec_blocks.2.0.conv2.convs.2.weight": { "torch": "dec_blocks.2.0.conv2.weight", "transform": "[:, :, 2]", "to_onnx": [ { "index": [ 2, 2 ] } ] }, "dec_blocks.2.0.norm2.weight": { "torch": "dec_blocks.2.0.norm2.weight", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.norm2.bias": { "torch": "dec_blocks.2.0.norm2.bias", "transform": "identity", "to_onnx": [] }, "dec_blocks.2.0.skip.convs.0.weight": { "torch": "dec_blocks.2.0.skip.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "dec_blocks.2.0.skip.convs.0.bias": { "torch": "dec_blocks.2.0.skip.bias", "transform": "identity", "to_onnx": [] }, "final_conv.convs.0.weight": { "torch": "final_conv.weight", "transform": "[:, :, 0]", "to_onnx": [ { "index": [ 2, 0 ] } ] }, "final_conv.convs.0.bias": { "torch": "final_conv.bias", "transform": "identity", "to_onnx": [] } }