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_61 /finetune /torch_params.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
13 kB
{
"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"
}
]
}
}