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_53 /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.7 kB
{
"encoder0.conv1.convs.0.weight": {
"torch": "encoder0.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"encoder0.conv1.convs.0.bias": {
"torch": "encoder0.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"encoder0.conv1.convs.1.weight": {
"torch": "encoder0.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"encoder0.conv1.convs.2.weight": {
"torch": "encoder0.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"encoder0.bn1.weight": {
"torch": "encoder0.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"encoder0.bn1.bias": {
"torch": "encoder0.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"encoder0.bn1.running_mean": {
"torch": "encoder0.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"encoder0.bn1.running_var": {
"torch": "encoder0.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"encoder0.conv2.convs.0.weight": {
"torch": "encoder0.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"encoder0.conv2.convs.0.bias": {
"torch": "encoder0.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"encoder0.conv2.convs.1.weight": {
"torch": "encoder0.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"encoder0.conv2.convs.2.weight": {
"torch": "encoder0.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"encoder0.bn2.weight": {
"torch": "encoder0.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"encoder0.bn2.bias": {
"torch": "encoder0.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"encoder0.bn2.running_mean": {
"torch": "encoder0.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"encoder0.bn2.running_var": {
"torch": "encoder0.bn2.running_var",
"transform": "identity",
"to_onnx": []
},
"encoder1.conv1.convs.0.weight": {
"torch": "encoder1.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"encoder1.conv1.convs.0.bias": {
"torch": "encoder1.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"encoder1.conv1.convs.1.weight": {
"torch": "encoder1.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"encoder1.conv1.convs.2.weight": {
"torch": "encoder1.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"encoder1.bn1.weight": {
"torch": "encoder1.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"encoder1.bn1.bias": {
"torch": "encoder1.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"encoder1.bn1.running_mean": {
"torch": "encoder1.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"encoder1.bn1.running_var": {
"torch": "encoder1.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"encoder1.conv2.convs.0.weight": {
"torch": "encoder1.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"encoder1.conv2.convs.0.bias": {
"torch": "encoder1.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"encoder1.conv2.convs.1.weight": {
"torch": "encoder1.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"encoder1.conv2.convs.2.weight": {
"torch": "encoder1.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"encoder1.bn2.weight": {
"torch": "encoder1.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"encoder1.bn2.bias": {
"torch": "encoder1.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"encoder1.bn2.running_mean": {
"torch": "encoder1.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"encoder1.bn2.running_var": {
"torch": "encoder1.bn2.running_var",
"transform": "identity",
"to_onnx": []
},
"encoder2.conv1.convs.0.weight": {
"torch": "encoder2.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"encoder2.conv1.convs.0.bias": {
"torch": "encoder2.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"encoder2.conv1.convs.1.weight": {
"torch": "encoder2.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"encoder2.conv1.convs.2.weight": {
"torch": "encoder2.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"encoder2.bn1.weight": {
"torch": "encoder2.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"encoder2.bn1.bias": {
"torch": "encoder2.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"encoder2.bn1.running_mean": {
"torch": "encoder2.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"encoder2.bn1.running_var": {
"torch": "encoder2.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"encoder2.conv2.convs.0.weight": {
"torch": "encoder2.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"encoder2.conv2.convs.0.bias": {
"torch": "encoder2.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"encoder2.conv2.convs.1.weight": {
"torch": "encoder2.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"encoder2.conv2.convs.2.weight": {
"torch": "encoder2.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"encoder2.bn2.weight": {
"torch": "encoder2.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"encoder2.bn2.bias": {
"torch": "encoder2.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"encoder2.bn2.running_mean": {
"torch": "encoder2.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"encoder2.bn2.running_var": {
"torch": "encoder2.bn2.running_var",
"transform": "identity",
"to_onnx": []
},
"bottleneck.conv1.convs.0.weight": {
"torch": "bottleneck.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"bottleneck.conv1.convs.0.bias": {
"torch": "bottleneck.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"bottleneck.conv1.convs.1.weight": {
"torch": "bottleneck.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"bottleneck.conv1.convs.2.weight": {
"torch": "bottleneck.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"bottleneck.bn1.weight": {
"torch": "bottleneck.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"bottleneck.bn1.bias": {
"torch": "bottleneck.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"bottleneck.bn1.running_mean": {
"torch": "bottleneck.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"bottleneck.bn1.running_var": {
"torch": "bottleneck.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"bottleneck.conv2.convs.0.weight": {
"torch": "bottleneck.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"bottleneck.conv2.convs.0.bias": {
"torch": "bottleneck.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"bottleneck.conv2.convs.1.weight": {
"torch": "bottleneck.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"bottleneck.conv2.convs.2.weight": {
"torch": "bottleneck.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"bottleneck.bn2.weight": {
"torch": "bottleneck.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"bottleneck.bn2.bias": {
"torch": "bottleneck.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"bottleneck.bn2.running_mean": {
"torch": "bottleneck.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"bottleneck.bn2.running_var": {
"torch": "bottleneck.bn2.running_var",
"transform": "identity",
"to_onnx": []
},
"upconv2.conv.convs.0.weight": {
"torch": "upconv2.weight",
"transform": "permute(1,2,3,4,0).reshape(Cout*8, Cin, 1, 1)",
"to_onnx": [
{
"permute": [
1,
2,
3,
4,
0
]
},
{
"reshape": [
1024,
256,
1,
1
]
}
]
},
"upconv2.conv.convs.0.bias": {
"torch": "upconv2.bias",
"transform": "repeat_interleave(8)",
"to_onnx": [
{
"repeat_interleave": [
0,
8
]
}
]
},
"decoder2.conv1.convs.0.weight": {
"torch": "decoder2.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"decoder2.conv1.convs.0.bias": {
"torch": "decoder2.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"decoder2.conv1.convs.1.weight": {
"torch": "decoder2.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"decoder2.conv1.convs.2.weight": {
"torch": "decoder2.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"decoder2.bn1.weight": {
"torch": "decoder2.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"decoder2.bn1.bias": {
"torch": "decoder2.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"decoder2.bn1.running_mean": {
"torch": "decoder2.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"decoder2.bn1.running_var": {
"torch": "decoder2.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"decoder2.conv2.convs.0.weight": {
"torch": "decoder2.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"decoder2.conv2.convs.0.bias": {
"torch": "decoder2.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"decoder2.conv2.convs.1.weight": {
"torch": "decoder2.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"decoder2.conv2.convs.2.weight": {
"torch": "decoder2.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"decoder2.bn2.weight": {
"torch": "decoder2.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"decoder2.bn2.bias": {
"torch": "decoder2.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"decoder2.bn2.running_mean": {
"torch": "decoder2.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"decoder2.bn2.running_var": {
"torch": "decoder2.bn2.running_var",
"transform": "identity",
"to_onnx": []
},
"upconv1.conv.convs.0.weight": {
"torch": "upconv1.weight",
"transform": "permute(1,2,3,4,0).reshape(Cout*8, Cin, 1, 1)",
"to_onnx": [
{
"permute": [
1,
2,
3,
4,
0
]
},
{
"reshape": [
512,
128,
1,
1
]
}
]
},
"upconv1.conv.convs.0.bias": {
"torch": "upconv1.bias",
"transform": "repeat_interleave(8)",
"to_onnx": [
{
"repeat_interleave": [
0,
8
]
}
]
},
"decoder1.conv1.convs.0.weight": {
"torch": "decoder1.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"decoder1.conv1.convs.0.bias": {
"torch": "decoder1.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"decoder1.conv1.convs.1.weight": {
"torch": "decoder1.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"decoder1.conv1.convs.2.weight": {
"torch": "decoder1.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"decoder1.bn1.weight": {
"torch": "decoder1.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"decoder1.bn1.bias": {
"torch": "decoder1.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"decoder1.bn1.running_mean": {
"torch": "decoder1.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"decoder1.bn1.running_var": {
"torch": "decoder1.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"decoder1.conv2.convs.0.weight": {
"torch": "decoder1.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"decoder1.conv2.convs.0.bias": {
"torch": "decoder1.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"decoder1.conv2.convs.1.weight": {
"torch": "decoder1.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"decoder1.conv2.convs.2.weight": {
"torch": "decoder1.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"decoder1.bn2.weight": {
"torch": "decoder1.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"decoder1.bn2.bias": {
"torch": "decoder1.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"decoder1.bn2.running_mean": {
"torch": "decoder1.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"decoder1.bn2.running_var": {
"torch": "decoder1.bn2.running_var",
"transform": "identity",
"to_onnx": []
},
"upconv0.conv.convs.0.weight": {
"torch": "upconv0.weight",
"transform": "permute(1,2,3,4,0).reshape(Cout*8, Cin, 1, 1)",
"to_onnx": [
{
"permute": [
1,
2,
3,
4,
0
]
},
{
"reshape": [
256,
64,
1,
1
]
}
]
},
"upconv0.conv.convs.0.bias": {
"torch": "upconv0.bias",
"transform": "repeat_interleave(8)",
"to_onnx": [
{
"repeat_interleave": [
0,
8
]
}
]
},
"decoder0.conv1.convs.0.weight": {
"torch": "decoder0.conv1.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"decoder0.conv1.convs.0.bias": {
"torch": "decoder0.conv1.bias",
"transform": "identity",
"to_onnx": []
},
"decoder0.conv1.convs.1.weight": {
"torch": "decoder0.conv1.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"decoder0.conv1.convs.2.weight": {
"torch": "decoder0.conv1.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"decoder0.bn1.weight": {
"torch": "decoder0.bn1.weight",
"transform": "identity",
"to_onnx": []
},
"decoder0.bn1.bias": {
"torch": "decoder0.bn1.bias",
"transform": "identity",
"to_onnx": []
},
"decoder0.bn1.running_mean": {
"torch": "decoder0.bn1.running_mean",
"transform": "identity",
"to_onnx": []
},
"decoder0.bn1.running_var": {
"torch": "decoder0.bn1.running_var",
"transform": "identity",
"to_onnx": []
},
"decoder0.conv2.convs.0.weight": {
"torch": "decoder0.conv2.weight",
"transform": "[:, :, 0]",
"to_onnx": [
{
"index": [
2,
0
]
}
]
},
"decoder0.conv2.convs.0.bias": {
"torch": "decoder0.conv2.bias",
"transform": "identity",
"to_onnx": []
},
"decoder0.conv2.convs.1.weight": {
"torch": "decoder0.conv2.weight",
"transform": "[:, :, 1]",
"to_onnx": [
{
"index": [
2,
1
]
}
]
},
"decoder0.conv2.convs.2.weight": {
"torch": "decoder0.conv2.weight",
"transform": "[:, :, 2]",
"to_onnx": [
{
"index": [
2,
2
]
}
]
},
"decoder0.bn2.weight": {
"torch": "decoder0.bn2.weight",
"transform": "identity",
"to_onnx": []
},
"decoder0.bn2.bias": {
"torch": "decoder0.bn2.bias",
"transform": "identity",
"to_onnx": []
},
"decoder0.bn2.running_mean": {
"torch": "decoder0.bn2.running_mean",
"transform": "identity",
"to_onnx": []
},
"decoder0.bn2.running_var": {
"torch": "decoder0.bn2.running_var",
"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": []
}
}