Hugging Face's logo

cellmap
/
onnx-web-models

ONNX
electron-microscopy
segmentation
cellmap
webgpu
onnxruntime-web
Model card Files
xet
 Community
onnx-web-models / setup_67 /finetune /weight_map.json
zouinkhim's picture
zouinkhim
3D full finetuning: NDHWC train_graph.json + torch_params.json, structured weight_map to_onnx, 3D runtime; 2D spike peak_mb corrected
3331842 verified
Raw History Blame Contribute Delete
16.1 kB
{
"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": []
}
}