{ "format": "cellmap-train-graph/1", "layout": "NDHWC", "input": "input", "output": "output", "context": 88, "params": { "stem.conv1.weight": { "shape": [ 40, 1, 3, 3, 3 ], "trainable": true }, "stem.conv1.bias": { "shape": [ 40 ], "trainable": true }, "stem.norm1.weight": { "shape": [ 40 ], "trainable": true }, "stem.norm1.bias": { "shape": [ 40 ], "trainable": true }, "stem.conv2.weight": { "shape": [ 40, 40, 3, 3, 3 ], "trainable": true }, "stem.conv2.bias": { "shape": [ 40 ], "trainable": true }, "stem.norm2.weight": { "shape": [ 40 ], "trainable": true }, "stem.norm2.bias": { "shape": [ 40 ], "trainable": true }, "stem.skip.weight": { "shape": [ 40, 1, 1, 1, 1 ], "trainable": true }, "stem.skip.bias": { "shape": [ 40 ], "trainable": true }, "downs.0.conv.weight": { "shape": [ 80, 40, 2, 2, 2 ], "trainable": true }, "downs.0.conv.bias": { "shape": [ 80 ], "trainable": true }, "downs.1.conv.weight": { "shape": [ 160, 80, 2, 2, 2 ], "trainable": true }, "downs.1.conv.bias": { "shape": [ 160 ], "trainable": true }, "downs.2.conv.weight": { "shape": [ 320, 160, 2, 2, 2 ], "trainable": true }, "downs.2.conv.bias": { "shape": [ 320 ], "trainable": true }, "enc_blocks.0.0.conv1.weight": { "shape": [ 80, 80, 3, 3, 3 ], "trainable": true }, "enc_blocks.0.0.conv1.bias": { "shape": [ 80 ], "trainable": true }, "enc_blocks.0.0.norm1.weight": { "shape": [ 80 ], "trainable": true }, "enc_blocks.0.0.norm1.bias": { "shape": [ 80 ], "trainable": true }, "enc_blocks.0.0.conv2.weight": { "shape": [ 80, 80, 3, 3, 3 ], "trainable": true }, "enc_blocks.0.0.conv2.bias": { "shape": [ 80 ], "trainable": true }, "enc_blocks.0.0.norm2.weight": { "shape": [ 80 ], "trainable": true }, "enc_blocks.0.0.norm2.bias": { "shape": [ 80 ], "trainable": true }, "enc_blocks.1.0.conv1.weight": { "shape": [ 160, 160, 3, 3, 3 ], "trainable": true }, "enc_blocks.1.0.conv1.bias": { "shape": [ 160 ], "trainable": true }, "enc_blocks.1.0.norm1.weight": { "shape": [ 160 ], "trainable": true }, "enc_blocks.1.0.norm1.bias": { "shape": [ 160 ], "trainable": true }, "enc_blocks.1.0.conv2.weight": { "shape": [ 160, 160, 3, 3, 3 ], "trainable": true }, "enc_blocks.1.0.conv2.bias": { "shape": [ 160 ], "trainable": true }, "enc_blocks.1.0.norm2.weight": { "shape": [ 160 ], "trainable": true }, "enc_blocks.1.0.norm2.bias": { "shape": [ 160 ], "trainable": true }, "enc_blocks.2.0.conv1.weight": { "shape": [ 320, 320, 3, 3, 3 ], "trainable": true }, "enc_blocks.2.0.conv1.bias": { "shape": [ 320 ], "trainable": true }, "enc_blocks.2.0.norm1.weight": { "shape": [ 320 ], "trainable": true }, "enc_blocks.2.0.norm1.bias": { "shape": [ 320 ], "trainable": true }, "enc_blocks.2.0.conv2.weight": { "shape": [ 320, 320, 3, 3, 3 ], "trainable": true }, "enc_blocks.2.0.conv2.bias": { "shape": [ 320 ], "trainable": true }, "enc_blocks.2.0.norm2.weight": { "shape": [ 320 ], "trainable": true }, "enc_blocks.2.0.norm2.bias": { "shape": [ 320 ], "trainable": true }, "ups.0.reduce.weight": { "shape": [ 160, 320, 1, 1, 1 ], "trainable": true }, "ups.0.reduce.bias": { "shape": [ 160 ], "trainable": true }, "ups.1.reduce.weight": { "shape": [ 80, 160, 1, 1, 1 ], "trainable": true }, "ups.1.reduce.bias": { "shape": [ 80 ], "trainable": true }, "ups.2.reduce.weight": { "shape": [ 40, 80, 1, 1, 1 ], "trainable": true }, "ups.2.reduce.bias": { "shape": [ 40 ], "trainable": true }, "dec_blocks.0.0.conv1.weight": { "shape": [ 160, 320, 3, 3, 3 ], "trainable": true }, "dec_blocks.0.0.conv1.bias": { "shape": [ 160 ], "trainable": true }, "dec_blocks.0.0.norm1.weight": { "shape": [ 160 ], "trainable": true }, "dec_blocks.0.0.norm1.bias": { "shape": [ 160 ], "trainable": true }, "dec_blocks.0.0.conv2.weight": { "shape": [ 160, 160, 3, 3, 3 ], "trainable": true }, "dec_blocks.0.0.conv2.bias": { "shape": [ 160 ], "trainable": true }, "dec_blocks.0.0.norm2.weight": { "shape": [ 160 ], "trainable": true }, "dec_blocks.0.0.norm2.bias": { "shape": [ 160 ], "trainable": true }, "dec_blocks.0.0.skip.weight": { "shape": [ 160, 320, 1, 1, 1 ], "trainable": true }, "dec_blocks.0.0.skip.bias": { "shape": [ 160 ], "trainable": true }, "dec_blocks.1.0.conv1.weight": { "shape": [ 80, 160, 3, 3, 3 ], "trainable": true }, "dec_blocks.1.0.conv1.bias": { "shape": [ 80 ], "trainable": true }, "dec_blocks.1.0.norm1.weight": { "shape": [ 80 ], "trainable": true }, "dec_blocks.1.0.norm1.bias": { "shape": [ 80 ], "trainable": true }, "dec_blocks.1.0.conv2.weight": { "shape": [ 80, 80, 3, 3, 3 ], "trainable": true }, "dec_blocks.1.0.conv2.bias": { "shape": [ 80 ], "trainable": true }, "dec_blocks.1.0.norm2.weight": { "shape": [ 80 ], "trainable": true }, "dec_blocks.1.0.norm2.bias": { "shape": [ 80 ], "trainable": true }, "dec_blocks.1.0.skip.weight": { "shape": [ 80, 160, 1, 1, 1 ], "trainable": true }, "dec_blocks.1.0.skip.bias": { "shape": [ 80 ], "trainable": true }, "dec_blocks.2.0.conv1.weight": { "shape": [ 40, 80, 3, 3, 3 ], "trainable": true }, "dec_blocks.2.0.conv1.bias": { "shape": [ 40 ], "trainable": true }, "dec_blocks.2.0.norm1.weight": { "shape": [ 40 ], "trainable": true }, "dec_blocks.2.0.norm1.bias": { "shape": [ 40 ], "trainable": true }, "dec_blocks.2.0.conv2.weight": { "shape": [ 40, 40, 3, 3, 3 ], "trainable": true }, "dec_blocks.2.0.conv2.bias": { "shape": [ 40 ], "trainable": true }, "dec_blocks.2.0.norm2.weight": { "shape": [ 40 ], "trainable": true }, "dec_blocks.2.0.norm2.bias": { "shape": [ 40 ], "trainable": true }, "dec_blocks.2.0.skip.weight": { "shape": [ 40, 80, 1, 1, 1 ], "trainable": true }, "dec_blocks.2.0.skip.bias": { "shape": [ 40 ], "trainable": true }, "final_conv.weight": { "shape": [ 1, 40, 1, 1, 1 ], "trainable": true }, "final_conv.bias": { "shape": [ 1 ], "trainable": true } }, "ops": [ { "op": "conv", "x": "input", "y": "/stem/skip/Conv_output_0", "w": "stem.skip.weight", "b": "stem.skip.bias", "stride": [ 1, 1, 1 ] }, { "op": "conv", "x": "input", "y": "/stem/conv1/Conv_output_0", "w": "stem.conv1.weight", "b": "stem.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/stem/conv1/Conv_output_0", "y": "/stem/norm1/InstanceNormalization_output_0", "scale": "stem.norm1.weight", "bias": "stem.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/stem/norm1/InstanceNormalization_output_0", "y": "/stem/act/Relu_output_0" }, { "op": "conv", "x": "/stem/act/Relu_output_0", "y": "/stem/conv2/Conv_output_0", "w": "stem.conv2.weight", "b": "stem.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/stem/conv2/Conv_output_0", "y": "/stem/norm2/InstanceNormalization_output_0", "scale": "stem.norm2.weight", "bias": "stem.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/stem/skip/Conv_output_0", "y": "/stem/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/stem/Slice_output_0", "y": "/stem/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/stem/Slice_1_output_0", "y": "/stem/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/stem/norm2/InstanceNormalization_output_0", "x2": "/stem/Slice_2_output_0", "y": "/stem/Add_output_0" }, { "op": "relu", "x": "/stem/Add_output_0", "y": "/stem/act_1/Relu_output_0" }, { "op": "conv", "x": "/stem/act_1/Relu_output_0", "y": "/downs.0/conv/Conv_output_0", "w": "downs.0.conv.weight", "b": "downs.0.conv.bias", "stride": [ 2, 2, 2 ] }, { "op": "conv", "x": "/downs.0/conv/Conv_output_0", "y": "/enc_blocks.0.0/conv1/Conv_output_0", "w": "enc_blocks.0.0.conv1.weight", "b": "enc_blocks.0.0.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/enc_blocks.0.0/conv1/Conv_output_0", "y": "/enc_blocks.0.0/norm1/InstanceNormalization_output_0", "scale": "enc_blocks.0.0.norm1.weight", "bias": "enc_blocks.0.0.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/enc_blocks.0.0/norm1/InstanceNormalization_output_0", "y": "/enc_blocks.0.0/act/Relu_output_0" }, { "op": "conv", "x": "/enc_blocks.0.0/act/Relu_output_0", "y": "/enc_blocks.0.0/conv2/Conv_output_0", "w": "enc_blocks.0.0.conv2.weight", "b": "enc_blocks.0.0.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/enc_blocks.0.0/conv2/Conv_output_0", "y": "/enc_blocks.0.0/norm2/InstanceNormalization_output_0", "scale": "enc_blocks.0.0.norm2.weight", "bias": "enc_blocks.0.0.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/downs.0/conv/Conv_output_0", "y": "/enc_blocks.0.0/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/enc_blocks.0.0/Slice_output_0", "y": "/enc_blocks.0.0/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/enc_blocks.0.0/Slice_1_output_0", "y": "/enc_blocks.0.0/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/enc_blocks.0.0/norm2/InstanceNormalization_output_0", "x2": "/enc_blocks.0.0/Slice_2_output_0", "y": "/enc_blocks.0.0/Add_output_0" }, { "op": "relu", "x": "/enc_blocks.0.0/Add_output_0", "y": "/enc_blocks.0.0/act_1/Relu_output_0" }, { "op": "conv", "x": "/enc_blocks.0.0/act_1/Relu_output_0", "y": "/downs.1/conv/Conv_output_0", "w": "downs.1.conv.weight", "b": "downs.1.conv.bias", "stride": [ 2, 2, 2 ] }, { "op": "conv", "x": "/downs.1/conv/Conv_output_0", "y": "/enc_blocks.1.0/conv1/Conv_output_0", "w": "enc_blocks.1.0.conv1.weight", "b": "enc_blocks.1.0.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/enc_blocks.1.0/conv1/Conv_output_0", "y": "/enc_blocks.1.0/norm1/InstanceNormalization_output_0", "scale": "enc_blocks.1.0.norm1.weight", "bias": "enc_blocks.1.0.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/enc_blocks.1.0/norm1/InstanceNormalization_output_0", "y": "/enc_blocks.1.0/act/Relu_output_0" }, { "op": "conv", "x": "/enc_blocks.1.0/act/Relu_output_0", "y": "/enc_blocks.1.0/conv2/Conv_output_0", "w": "enc_blocks.1.0.conv2.weight", "b": "enc_blocks.1.0.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/enc_blocks.1.0/conv2/Conv_output_0", "y": "/enc_blocks.1.0/norm2/InstanceNormalization_output_0", "scale": "enc_blocks.1.0.norm2.weight", "bias": "enc_blocks.1.0.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/downs.1/conv/Conv_output_0", "y": "/enc_blocks.1.0/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/enc_blocks.1.0/Slice_output_0", "y": "/enc_blocks.1.0/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/enc_blocks.1.0/Slice_1_output_0", "y": "/enc_blocks.1.0/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/enc_blocks.1.0/norm2/InstanceNormalization_output_0", "x2": "/enc_blocks.1.0/Slice_2_output_0", "y": "/enc_blocks.1.0/Add_output_0" }, { "op": "relu", "x": "/enc_blocks.1.0/Add_output_0", "y": "/enc_blocks.1.0/act_1/Relu_output_0" }, { "op": "conv", "x": "/enc_blocks.1.0/act_1/Relu_output_0", "y": "/downs.2/conv/Conv_output_0", "w": "downs.2.conv.weight", "b": "downs.2.conv.bias", "stride": [ 2, 2, 2 ] }, { "op": "conv", "x": "/downs.2/conv/Conv_output_0", "y": "/enc_blocks.2.0/conv1/Conv_output_0", "w": "enc_blocks.2.0.conv1.weight", "b": "enc_blocks.2.0.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/enc_blocks.2.0/conv1/Conv_output_0", "y": "/enc_blocks.2.0/norm1/InstanceNormalization_output_0", "scale": "enc_blocks.2.0.norm1.weight", "bias": "enc_blocks.2.0.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/enc_blocks.2.0/norm1/InstanceNormalization_output_0", "y": "/enc_blocks.2.0/act/Relu_output_0" }, { "op": "conv", "x": "/enc_blocks.2.0/act/Relu_output_0", "y": "/enc_blocks.2.0/conv2/Conv_output_0", "w": "enc_blocks.2.0.conv2.weight", "b": "enc_blocks.2.0.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/enc_blocks.2.0/conv2/Conv_output_0", "y": "/enc_blocks.2.0/norm2/InstanceNormalization_output_0", "scale": "enc_blocks.2.0.norm2.weight", "bias": "enc_blocks.2.0.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/downs.2/conv/Conv_output_0", "y": "/enc_blocks.2.0/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/enc_blocks.2.0/Slice_output_0", "y": "/enc_blocks.2.0/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/enc_blocks.2.0/Slice_1_output_0", "y": "/enc_blocks.2.0/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/enc_blocks.2.0/norm2/InstanceNormalization_output_0", "x2": "/enc_blocks.2.0/Slice_2_output_0", "y": "/enc_blocks.2.0/Add_output_0" }, { "op": "relu", "x": "/enc_blocks.2.0/Add_output_0", "y": "/enc_blocks.2.0/act_1/Relu_output_0" }, { "op": "upsample", "x": "/enc_blocks.2.0/act_1/Relu_output_0", "y": "/ups.0/Resize_output_0", "factor": 2 }, { "op": "conv", "x": "/ups.0/Resize_output_0", "y": "/ups.0/reduce/Conv_output_0", "w": "ups.0.reduce.weight", "b": "ups.0.reduce.bias", "stride": [ 1, 1, 1 ] }, { "op": "crop", "x": "/enc_blocks.1.0/act_1/Relu_output_0", "y": "/Slice_output_0", "axis": "d", "start": 4, "end_margin": 4 }, { "op": "crop", "x": "/Slice_output_0", "y": "/Slice_1_output_0", "axis": "h", "start": 4, "end_margin": 4 }, { "op": "crop", "x": "/Slice_1_output_0", "y": "/Slice_2_output_0", "axis": "w", "start": 4, "end_margin": 4 }, { "op": "concat", "xs": [ "/ups.0/reduce/Conv_output_0", "/Slice_2_output_0" ], "y": "/Concat_output_0" }, { "op": "conv", "x": "/Concat_output_0", "y": "/dec_blocks.0.0/skip/Conv_output_0", "w": "dec_blocks.0.0.skip.weight", "b": "dec_blocks.0.0.skip.bias", "stride": [ 1, 1, 1 ] }, { "op": "conv", "x": "/Concat_output_0", "y": "/dec_blocks.0.0/conv1/Conv_output_0", "w": "dec_blocks.0.0.conv1.weight", "b": "dec_blocks.0.0.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/dec_blocks.0.0/conv1/Conv_output_0", "y": "/dec_blocks.0.0/norm1/InstanceNormalization_output_0", "scale": "dec_blocks.0.0.norm1.weight", "bias": "dec_blocks.0.0.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/dec_blocks.0.0/norm1/InstanceNormalization_output_0", "y": "/dec_blocks.0.0/act/Relu_output_0" }, { "op": "conv", "x": "/dec_blocks.0.0/act/Relu_output_0", "y": "/dec_blocks.0.0/conv2/Conv_output_0", "w": "dec_blocks.0.0.conv2.weight", "b": "dec_blocks.0.0.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/dec_blocks.0.0/conv2/Conv_output_0", "y": "/dec_blocks.0.0/norm2/InstanceNormalization_output_0", "scale": "dec_blocks.0.0.norm2.weight", "bias": "dec_blocks.0.0.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/dec_blocks.0.0/skip/Conv_output_0", "y": "/dec_blocks.0.0/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/dec_blocks.0.0/Slice_output_0", "y": "/dec_blocks.0.0/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/dec_blocks.0.0/Slice_1_output_0", "y": "/dec_blocks.0.0/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/dec_blocks.0.0/norm2/InstanceNormalization_output_0", "x2": "/dec_blocks.0.0/Slice_2_output_0", "y": "/dec_blocks.0.0/Add_output_0" }, { "op": "relu", "x": "/dec_blocks.0.0/Add_output_0", "y": "/dec_blocks.0.0/act_1/Relu_output_0" }, { "op": "upsample", "x": "/dec_blocks.0.0/act_1/Relu_output_0", "y": "/ups.1/Resize_output_0", "factor": 2 }, { "op": "conv", "x": "/ups.1/Resize_output_0", "y": "/ups.1/reduce/Conv_output_0", "w": "ups.1.reduce.weight", "b": "ups.1.reduce.bias", "stride": [ 1, 1, 1 ] }, { "op": "crop", "x": "/enc_blocks.0.0/act_1/Relu_output_0", "y": "/Slice_3_output_0", "axis": "d", "start": 16, "end_margin": 16 }, { "op": "crop", "x": "/Slice_3_output_0", "y": "/Slice_4_output_0", "axis": "h", "start": 16, "end_margin": 16 }, { "op": "crop", "x": "/Slice_4_output_0", "y": "/Slice_5_output_0", "axis": "w", "start": 16, "end_margin": 16 }, { "op": "concat", "xs": [ "/ups.1/reduce/Conv_output_0", "/Slice_5_output_0" ], "y": "/Concat_1_output_0" }, { "op": "conv", "x": "/Concat_1_output_0", "y": "/dec_blocks.1.0/skip/Conv_output_0", "w": "dec_blocks.1.0.skip.weight", "b": "dec_blocks.1.0.skip.bias", "stride": [ 1, 1, 1 ] }, { "op": "conv", "x": "/Concat_1_output_0", "y": "/dec_blocks.1.0/conv1/Conv_output_0", "w": "dec_blocks.1.0.conv1.weight", "b": "dec_blocks.1.0.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/dec_blocks.1.0/conv1/Conv_output_0", "y": "/dec_blocks.1.0/norm1/InstanceNormalization_output_0", "scale": "dec_blocks.1.0.norm1.weight", "bias": "dec_blocks.1.0.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/dec_blocks.1.0/norm1/InstanceNormalization_output_0", "y": "/dec_blocks.1.0/act/Relu_output_0" }, { "op": "conv", "x": "/dec_blocks.1.0/act/Relu_output_0", "y": "/dec_blocks.1.0/conv2/Conv_output_0", "w": "dec_blocks.1.0.conv2.weight", "b": "dec_blocks.1.0.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/dec_blocks.1.0/conv2/Conv_output_0", "y": "/dec_blocks.1.0/norm2/InstanceNormalization_output_0", "scale": "dec_blocks.1.0.norm2.weight", "bias": "dec_blocks.1.0.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/dec_blocks.1.0/skip/Conv_output_0", "y": "/dec_blocks.1.0/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/dec_blocks.1.0/Slice_output_0", "y": "/dec_blocks.1.0/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/dec_blocks.1.0/Slice_1_output_0", "y": "/dec_blocks.1.0/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/dec_blocks.1.0/norm2/InstanceNormalization_output_0", "x2": "/dec_blocks.1.0/Slice_2_output_0", "y": "/dec_blocks.1.0/Add_output_0" }, { "op": "relu", "x": "/dec_blocks.1.0/Add_output_0", "y": "/dec_blocks.1.0/act_1/Relu_output_0" }, { "op": "upsample", "x": "/dec_blocks.1.0/act_1/Relu_output_0", "y": "/ups.2/Resize_output_0", "factor": 2 }, { "op": "conv", "x": "/ups.2/Resize_output_0", "y": "/ups.2/reduce/Conv_output_0", "w": "ups.2.reduce.weight", "b": "ups.2.reduce.bias", "stride": [ 1, 1, 1 ] }, { "op": "crop", "x": "/stem/act_1/Relu_output_0", "y": "/Slice_6_output_0", "axis": "d", "start": 40, "end_margin": 40 }, { "op": "crop", "x": "/Slice_6_output_0", "y": "/Slice_7_output_0", "axis": "h", "start": 40, "end_margin": 40 }, { "op": "crop", "x": "/Slice_7_output_0", "y": "/Slice_8_output_0", "axis": "w", "start": 40, "end_margin": 40 }, { "op": "concat", "xs": [ "/ups.2/reduce/Conv_output_0", "/Slice_8_output_0" ], "y": "/Concat_2_output_0" }, { "op": "conv", "x": "/Concat_2_output_0", "y": "/dec_blocks.2.0/skip/Conv_output_0", "w": "dec_blocks.2.0.skip.weight", "b": "dec_blocks.2.0.skip.bias", "stride": [ 1, 1, 1 ] }, { "op": "conv", "x": "/Concat_2_output_0", "y": "/dec_blocks.2.0/conv1/Conv_output_0", "w": "dec_blocks.2.0.conv1.weight", "b": "dec_blocks.2.0.conv1.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/dec_blocks.2.0/conv1/Conv_output_0", "y": "/dec_blocks.2.0/norm1/InstanceNormalization_output_0", "scale": "dec_blocks.2.0.norm1.weight", "bias": "dec_blocks.2.0.norm1.bias", "eps": 9.999999747378752e-06 }, { "op": "relu", "x": "/dec_blocks.2.0/norm1/InstanceNormalization_output_0", "y": "/dec_blocks.2.0/act/Relu_output_0" }, { "op": "conv", "x": "/dec_blocks.2.0/act/Relu_output_0", "y": "/dec_blocks.2.0/conv2/Conv_output_0", "w": "dec_blocks.2.0.conv2.weight", "b": "dec_blocks.2.0.conv2.bias", "stride": [ 1, 1, 1 ] }, { "op": "inorm", "x": "/dec_blocks.2.0/conv2/Conv_output_0", "y": "/dec_blocks.2.0/norm2/InstanceNormalization_output_0", "scale": "dec_blocks.2.0.norm2.weight", "bias": "dec_blocks.2.0.norm2.bias", "eps": 9.999999747378752e-06 }, { "op": "crop", "x": "/dec_blocks.2.0/skip/Conv_output_0", "y": "/dec_blocks.2.0/Slice_output_0", "axis": "d", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/dec_blocks.2.0/Slice_output_0", "y": "/dec_blocks.2.0/Slice_1_output_0", "axis": "h", "start": 2, "end_margin": 2 }, { "op": "crop", "x": "/dec_blocks.2.0/Slice_1_output_0", "y": "/dec_blocks.2.0/Slice_2_output_0", "axis": "w", "start": 2, "end_margin": 2 }, { "op": "add", "x": "/dec_blocks.2.0/norm2/InstanceNormalization_output_0", "x2": "/dec_blocks.2.0/Slice_2_output_0", "y": "/dec_blocks.2.0/Add_output_0" }, { "op": "relu", "x": "/dec_blocks.2.0/Add_output_0", "y": "/dec_blocks.2.0/act_1/Relu_output_0" }, { "op": "conv", "x": "/dec_blocks.2.0/act_1/Relu_output_0", "y": "output", "w": "final_conv.weight", "b": "final_conv.bias", "stride": [ 1, 1, 1 ] } ] }