onnx-web-models / setup_55 /finetune /train_graph.json
zouinkhim's picture
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
20 kB
{
"format": "cellmap-train-graph/1",
"layout": "NDHWC",
"input": "input",
"output": "output",
"context": 88,
"params": {
"encoder0.conv1.weight": {
"shape": [
32,
1,
3,
3,
3
],
"trainable": true
},
"encoder0.conv1.bias": {
"shape": [
32
],
"trainable": true
},
"encoder0.bn1.weight": {
"shape": [
32
],
"trainable": true
},
"encoder0.bn1.bias": {
"shape": [
32
],
"trainable": true
},
"encoder0.bn1.running_mean": {
"shape": [
32
],
"trainable": false
},
"encoder0.bn1.running_var": {
"shape": [
32
],
"trainable": false
},
"encoder0.conv2.weight": {
"shape": [
32,
32,
3,
3,
3
],
"trainable": true
},
"encoder0.conv2.bias": {
"shape": [
32
],
"trainable": true
},
"encoder0.bn2.weight": {
"shape": [
32
],
"trainable": true
},
"encoder0.bn2.bias": {
"shape": [
32
],
"trainable": true
},
"encoder0.bn2.running_mean": {
"shape": [
32
],
"trainable": false
},
"encoder0.bn2.running_var": {
"shape": [
32
],
"trainable": false
},
"encoder1.conv1.weight": {
"shape": [
64,
32,
3,
3,
3
],
"trainable": true
},
"encoder1.conv1.bias": {
"shape": [
64
],
"trainable": true
},
"encoder1.bn1.weight": {
"shape": [
64
],
"trainable": true
},
"encoder1.bn1.bias": {
"shape": [
64
],
"trainable": true
},
"encoder1.bn1.running_mean": {
"shape": [
64
],
"trainable": false
},
"encoder1.bn1.running_var": {
"shape": [
64
],
"trainable": false
},
"encoder1.conv2.weight": {
"shape": [
64,
64,
3,
3,
3
],
"trainable": true
},
"encoder1.conv2.bias": {
"shape": [
64
],
"trainable": true
},
"encoder1.bn2.weight": {
"shape": [
64
],
"trainable": true
},
"encoder1.bn2.bias": {
"shape": [
64
],
"trainable": true
},
"encoder1.bn2.running_mean": {
"shape": [
64
],
"trainable": false
},
"encoder1.bn2.running_var": {
"shape": [
64
],
"trainable": false
},
"encoder2.conv1.weight": {
"shape": [
128,
64,
3,
3,
3
],
"trainable": true
},
"encoder2.conv1.bias": {
"shape": [
128
],
"trainable": true
},
"encoder2.bn1.weight": {
"shape": [
128
],
"trainable": true
},
"encoder2.bn1.bias": {
"shape": [
128
],
"trainable": true
},
"encoder2.bn1.running_mean": {
"shape": [
128
],
"trainable": false
},
"encoder2.bn1.running_var": {
"shape": [
128
],
"trainable": false
},
"encoder2.conv2.weight": {
"shape": [
128,
128,
3,
3,
3
],
"trainable": true
},
"encoder2.conv2.bias": {
"shape": [
128
],
"trainable": true
},
"encoder2.bn2.weight": {
"shape": [
128
],
"trainable": true
},
"encoder2.bn2.bias": {
"shape": [
128
],
"trainable": true
},
"encoder2.bn2.running_mean": {
"shape": [
128
],
"trainable": false
},
"encoder2.bn2.running_var": {
"shape": [
128
],
"trainable": false
},
"bottleneck.conv1.weight": {
"shape": [
256,
128,
3,
3,
3
],
"trainable": true
},
"bottleneck.conv1.bias": {
"shape": [
256
],
"trainable": true
},
"bottleneck.bn1.weight": {
"shape": [
256
],
"trainable": true
},
"bottleneck.bn1.bias": {
"shape": [
256
],
"trainable": true
},
"bottleneck.bn1.running_mean": {
"shape": [
256
],
"trainable": false
},
"bottleneck.bn1.running_var": {
"shape": [
256
],
"trainable": false
},
"bottleneck.conv2.weight": {
"shape": [
256,
256,
3,
3,
3
],
"trainable": true
},
"bottleneck.conv2.bias": {
"shape": [
256
],
"trainable": true
},
"bottleneck.bn2.weight": {
"shape": [
256
],
"trainable": true
},
"bottleneck.bn2.bias": {
"shape": [
256
],
"trainable": true
},
"bottleneck.bn2.running_mean": {
"shape": [
256
],
"trainable": false
},
"bottleneck.bn2.running_var": {
"shape": [
256
],
"trainable": false
},
"upconv2.weight": {
"shape": [
256,
128,
2,
2,
2
],
"trainable": true
},
"upconv2.bias": {
"shape": [
128
],
"trainable": true
},
"decoder2.conv1.weight": {
"shape": [
128,
256,
3,
3,
3
],
"trainable": true
},
"decoder2.conv1.bias": {
"shape": [
128
],
"trainable": true
},
"decoder2.bn1.weight": {
"shape": [
128
],
"trainable": true
},
"decoder2.bn1.bias": {
"shape": [
128
],
"trainable": true
},
"decoder2.bn1.running_mean": {
"shape": [
128
],
"trainable": false
},
"decoder2.bn1.running_var": {
"shape": [
128
],
"trainable": false
},
"decoder2.conv2.weight": {
"shape": [
128,
128,
3,
3,
3
],
"trainable": true
},
"decoder2.conv2.bias": {
"shape": [
128
],
"trainable": true
},
"decoder2.bn2.weight": {
"shape": [
128
],
"trainable": true
},
"decoder2.bn2.bias": {
"shape": [
128
],
"trainable": true
},
"decoder2.bn2.running_mean": {
"shape": [
128
],
"trainable": false
},
"decoder2.bn2.running_var": {
"shape": [
128
],
"trainable": false
},
"upconv1.weight": {
"shape": [
128,
64,
2,
2,
2
],
"trainable": true
},
"upconv1.bias": {
"shape": [
64
],
"trainable": true
},
"decoder1.conv1.weight": {
"shape": [
64,
128,
3,
3,
3
],
"trainable": true
},
"decoder1.conv1.bias": {
"shape": [
64
],
"trainable": true
},
"decoder1.bn1.weight": {
"shape": [
64
],
"trainable": true
},
"decoder1.bn1.bias": {
"shape": [
64
],
"trainable": true
},
"decoder1.bn1.running_mean": {
"shape": [
64
],
"trainable": false
},
"decoder1.bn1.running_var": {
"shape": [
64
],
"trainable": false
},
"decoder1.conv2.weight": {
"shape": [
64,
64,
3,
3,
3
],
"trainable": true
},
"decoder1.conv2.bias": {
"shape": [
64
],
"trainable": true
},
"decoder1.bn2.weight": {
"shape": [
64
],
"trainable": true
},
"decoder1.bn2.bias": {
"shape": [
64
],
"trainable": true
},
"decoder1.bn2.running_mean": {
"shape": [
64
],
"trainable": false
},
"decoder1.bn2.running_var": {
"shape": [
64
],
"trainable": false
},
"upconv0.weight": {
"shape": [
64,
32,
2,
2,
2
],
"trainable": true
},
"upconv0.bias": {
"shape": [
32
],
"trainable": true
},
"decoder0.conv1.weight": {
"shape": [
32,
64,
3,
3,
3
],
"trainable": true
},
"decoder0.conv1.bias": {
"shape": [
32
],
"trainable": true
},
"decoder0.bn1.weight": {
"shape": [
32
],
"trainable": true
},
"decoder0.bn1.bias": {
"shape": [
32
],
"trainable": true
},
"decoder0.bn1.running_mean": {
"shape": [
32
],
"trainable": false
},
"decoder0.bn1.running_var": {
"shape": [
32
],
"trainable": false
},
"decoder0.conv2.weight": {
"shape": [
32,
32,
3,
3,
3
],
"trainable": true
},
"decoder0.conv2.bias": {
"shape": [
32
],
"trainable": true
},
"decoder0.bn2.weight": {
"shape": [
32
],
"trainable": true
},
"decoder0.bn2.bias": {
"shape": [
32
],
"trainable": true
},
"decoder0.bn2.running_mean": {
"shape": [
32
],
"trainable": false
},
"decoder0.bn2.running_var": {
"shape": [
32
],
"trainable": false
},
"final_conv.weight": {
"shape": [
1,
32,
1,
1,
1
],
"trainable": true
},
"final_conv.bias": {
"shape": [
1
],
"trainable": true
}
},
"ops": [
{
"op": "conv",
"x": "input",
"y": "/encoder0/conv1/Conv_output_0",
"w": "encoder0.conv1.weight",
"b": "encoder0.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/encoder0/conv1/Conv_output_0",
"y": "/encoder0/bn1/BatchNormalization_output_0",
"scale": "encoder0.bn1.weight",
"bias": "encoder0.bn1.bias",
"mean": "encoder0.bn1.running_mean",
"var": "encoder0.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/encoder0/bn1/BatchNormalization_output_0",
"y": "/encoder0/act/Relu_output_0"
},
{
"op": "conv",
"x": "/encoder0/act/Relu_output_0",
"y": "/encoder0/conv2/Conv_output_0",
"w": "encoder0.conv2.weight",
"b": "encoder0.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/encoder0/conv2/Conv_output_0",
"y": "/encoder0/bn2/BatchNormalization_output_0",
"scale": "encoder0.bn2.weight",
"bias": "encoder0.bn2.bias",
"mean": "encoder0.bn2.running_mean",
"var": "encoder0.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/encoder0/bn2/BatchNormalization_output_0",
"y": "/encoder0/act_1/Relu_output_0"
},
{
"op": "maxpool",
"x": "/encoder0/act_1/Relu_output_0",
"y": "/pool1/MaxPool_output_0",
"k": [
2,
2,
2
],
"stride": [
2,
2,
2
]
},
{
"op": "conv",
"x": "/pool1/MaxPool_output_0",
"y": "/encoder1/conv1/Conv_output_0",
"w": "encoder1.conv1.weight",
"b": "encoder1.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/encoder1/conv1/Conv_output_0",
"y": "/encoder1/bn1/BatchNormalization_output_0",
"scale": "encoder1.bn1.weight",
"bias": "encoder1.bn1.bias",
"mean": "encoder1.bn1.running_mean",
"var": "encoder1.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/encoder1/bn1/BatchNormalization_output_0",
"y": "/encoder1/act/Relu_output_0"
},
{
"op": "conv",
"x": "/encoder1/act/Relu_output_0",
"y": "/encoder1/conv2/Conv_output_0",
"w": "encoder1.conv2.weight",
"b": "encoder1.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/encoder1/conv2/Conv_output_0",
"y": "/encoder1/bn2/BatchNormalization_output_0",
"scale": "encoder1.bn2.weight",
"bias": "encoder1.bn2.bias",
"mean": "encoder1.bn2.running_mean",
"var": "encoder1.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/encoder1/bn2/BatchNormalization_output_0",
"y": "/encoder1/act_1/Relu_output_0"
},
{
"op": "maxpool",
"x": "/encoder1/act_1/Relu_output_0",
"y": "/pool2/MaxPool_output_0",
"k": [
2,
2,
2
],
"stride": [
2,
2,
2
]
},
{
"op": "conv",
"x": "/pool2/MaxPool_output_0",
"y": "/encoder2/conv1/Conv_output_0",
"w": "encoder2.conv1.weight",
"b": "encoder2.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/encoder2/conv1/Conv_output_0",
"y": "/encoder2/bn1/BatchNormalization_output_0",
"scale": "encoder2.bn1.weight",
"bias": "encoder2.bn1.bias",
"mean": "encoder2.bn1.running_mean",
"var": "encoder2.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/encoder2/bn1/BatchNormalization_output_0",
"y": "/encoder2/act/Relu_output_0"
},
{
"op": "conv",
"x": "/encoder2/act/Relu_output_0",
"y": "/encoder2/conv2/Conv_output_0",
"w": "encoder2.conv2.weight",
"b": "encoder2.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/encoder2/conv2/Conv_output_0",
"y": "/encoder2/bn2/BatchNormalization_output_0",
"scale": "encoder2.bn2.weight",
"bias": "encoder2.bn2.bias",
"mean": "encoder2.bn2.running_mean",
"var": "encoder2.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/encoder2/bn2/BatchNormalization_output_0",
"y": "/encoder2/act_1/Relu_output_0"
},
{
"op": "maxpool",
"x": "/encoder2/act_1/Relu_output_0",
"y": "/pool3/MaxPool_output_0",
"k": [
2,
2,
2
],
"stride": [
2,
2,
2
]
},
{
"op": "conv",
"x": "/pool3/MaxPool_output_0",
"y": "/bottleneck/conv1/Conv_output_0",
"w": "bottleneck.conv1.weight",
"b": "bottleneck.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/bottleneck/conv1/Conv_output_0",
"y": "/bottleneck/bn1/BatchNormalization_output_0",
"scale": "bottleneck.bn1.weight",
"bias": "bottleneck.bn1.bias",
"mean": "bottleneck.bn1.running_mean",
"var": "bottleneck.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/bottleneck/bn1/BatchNormalization_output_0",
"y": "/bottleneck/act/Relu_output_0"
},
{
"op": "conv",
"x": "/bottleneck/act/Relu_output_0",
"y": "/bottleneck/conv2/Conv_output_0",
"w": "bottleneck.conv2.weight",
"b": "bottleneck.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/bottleneck/conv2/Conv_output_0",
"y": "/bottleneck/bn2/BatchNormalization_output_0",
"scale": "bottleneck.bn2.weight",
"bias": "bottleneck.bn2.bias",
"mean": "bottleneck.bn2.running_mean",
"var": "bottleneck.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/bottleneck/bn2/BatchNormalization_output_0",
"y": "/bottleneck/act_1/Relu_output_0"
},
{
"op": "conv_t",
"x": "/bottleneck/act_1/Relu_output_0",
"y": "/upconv2/ConvTranspose_output_0",
"w": "upconv2.weight",
"b": "upconv2.bias",
"stride": [
2,
2,
2
]
},
{
"op": "crop",
"x": "/encoder2/act_1/Relu_output_0",
"y": "/cc2/Slice_output_0",
"axis": "d",
"start": 4,
"end_margin": 4
},
{
"op": "crop",
"x": "/cc2/Slice_output_0",
"y": "/cc2/Slice_1_output_0",
"axis": "h",
"start": 4,
"end_margin": 4
},
{
"op": "crop",
"x": "/cc2/Slice_1_output_0",
"y": "/cc2/Slice_2_output_0",
"axis": "w",
"start": 4,
"end_margin": 4
},
{
"op": "concat",
"xs": [
"/cc2/Slice_2_output_0",
"/upconv2/ConvTranspose_output_0"
],
"y": "/cc2/Concat_output_0"
},
{
"op": "conv",
"x": "/cc2/Concat_output_0",
"y": "/decoder2/conv1/Conv_output_0",
"w": "decoder2.conv1.weight",
"b": "decoder2.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/decoder2/conv1/Conv_output_0",
"y": "/decoder2/bn1/BatchNormalization_output_0",
"scale": "decoder2.bn1.weight",
"bias": "decoder2.bn1.bias",
"mean": "decoder2.bn1.running_mean",
"var": "decoder2.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/decoder2/bn1/BatchNormalization_output_0",
"y": "/decoder2/act/Relu_output_0"
},
{
"op": "conv",
"x": "/decoder2/act/Relu_output_0",
"y": "/decoder2/conv2/Conv_output_0",
"w": "decoder2.conv2.weight",
"b": "decoder2.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/decoder2/conv2/Conv_output_0",
"y": "/decoder2/bn2/BatchNormalization_output_0",
"scale": "decoder2.bn2.weight",
"bias": "decoder2.bn2.bias",
"mean": "decoder2.bn2.running_mean",
"var": "decoder2.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/decoder2/bn2/BatchNormalization_output_0",
"y": "/decoder2/act_1/Relu_output_0"
},
{
"op": "conv_t",
"x": "/decoder2/act_1/Relu_output_0",
"y": "/upconv1/ConvTranspose_output_0",
"w": "upconv1.weight",
"b": "upconv1.bias",
"stride": [
2,
2,
2
]
},
{
"op": "crop",
"x": "/encoder1/act_1/Relu_output_0",
"y": "/cc1/Slice_output_0",
"axis": "d",
"start": 16,
"end_margin": 16
},
{
"op": "crop",
"x": "/cc1/Slice_output_0",
"y": "/cc1/Slice_1_output_0",
"axis": "h",
"start": 16,
"end_margin": 16
},
{
"op": "crop",
"x": "/cc1/Slice_1_output_0",
"y": "/cc1/Slice_2_output_0",
"axis": "w",
"start": 16,
"end_margin": 16
},
{
"op": "concat",
"xs": [
"/cc1/Slice_2_output_0",
"/upconv1/ConvTranspose_output_0"
],
"y": "/cc1/Concat_output_0"
},
{
"op": "conv",
"x": "/cc1/Concat_output_0",
"y": "/decoder1/conv1/Conv_output_0",
"w": "decoder1.conv1.weight",
"b": "decoder1.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/decoder1/conv1/Conv_output_0",
"y": "/decoder1/bn1/BatchNormalization_output_0",
"scale": "decoder1.bn1.weight",
"bias": "decoder1.bn1.bias",
"mean": "decoder1.bn1.running_mean",
"var": "decoder1.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/decoder1/bn1/BatchNormalization_output_0",
"y": "/decoder1/act/Relu_output_0"
},
{
"op": "conv",
"x": "/decoder1/act/Relu_output_0",
"y": "/decoder1/conv2/Conv_output_0",
"w": "decoder1.conv2.weight",
"b": "decoder1.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/decoder1/conv2/Conv_output_0",
"y": "/decoder1/bn2/BatchNormalization_output_0",
"scale": "decoder1.bn2.weight",
"bias": "decoder1.bn2.bias",
"mean": "decoder1.bn2.running_mean",
"var": "decoder1.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/decoder1/bn2/BatchNormalization_output_0",
"y": "/decoder1/act_1/Relu_output_0"
},
{
"op": "conv_t",
"x": "/decoder1/act_1/Relu_output_0",
"y": "/upconv0/ConvTranspose_output_0",
"w": "upconv0.weight",
"b": "upconv0.bias",
"stride": [
2,
2,
2
]
},
{
"op": "crop",
"x": "/encoder0/act_1/Relu_output_0",
"y": "/cc0/Slice_output_0",
"axis": "d",
"start": 40,
"end_margin": 40
},
{
"op": "crop",
"x": "/cc0/Slice_output_0",
"y": "/cc0/Slice_1_output_0",
"axis": "h",
"start": 40,
"end_margin": 40
},
{
"op": "crop",
"x": "/cc0/Slice_1_output_0",
"y": "/cc0/Slice_2_output_0",
"axis": "w",
"start": 40,
"end_margin": 40
},
{
"op": "concat",
"xs": [
"/cc0/Slice_2_output_0",
"/upconv0/ConvTranspose_output_0"
],
"y": "/cc0/Concat_output_0"
},
{
"op": "conv",
"x": "/cc0/Concat_output_0",
"y": "/decoder0/conv1/Conv_output_0",
"w": "decoder0.conv1.weight",
"b": "decoder0.conv1.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/decoder0/conv1/Conv_output_0",
"y": "/decoder0/bn1/BatchNormalization_output_0",
"scale": "decoder0.bn1.weight",
"bias": "decoder0.bn1.bias",
"mean": "decoder0.bn1.running_mean",
"var": "decoder0.bn1.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/decoder0/bn1/BatchNormalization_output_0",
"y": "/decoder0/act/Relu_output_0"
},
{
"op": "conv",
"x": "/decoder0/act/Relu_output_0",
"y": "/decoder0/conv2/Conv_output_0",
"w": "decoder0.conv2.weight",
"b": "decoder0.conv2.bias",
"stride": [
1,
1,
1
]
},
{
"op": "bn",
"x": "/decoder0/conv2/Conv_output_0",
"y": "/decoder0/bn2/BatchNormalization_output_0",
"scale": "decoder0.bn2.weight",
"bias": "decoder0.bn2.bias",
"mean": "decoder0.bn2.running_mean",
"var": "decoder0.bn2.running_var",
"eps": 9.999999747378752e-06
},
{
"op": "relu",
"x": "/decoder0/bn2/BatchNormalization_output_0",
"y": "/decoder0/act_1/Relu_output_0"
},
{
"op": "conv",
"x": "/decoder0/act_1/Relu_output_0",
"y": "output",
"w": "final_conv.weight",
"b": "final_conv.bias",
"stride": [
1,
1,
1
]
}
]
}