diff --git a/config.pbtxt b/config.pbtxt index 874dc91..350e5ea 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -11,7 +11,7 @@ input [ dims: [ -1, 28, 28, 1 ] }, { - name: "Conv2_input" + name: "Conv2_input2" data_type: TYPE_UINT16 dims: [ -1, 16 ] reshape { @@ -25,17 +25,25 @@ input [ reshape { shape: [ 123 ] } + }, + { + name: "Conv4_input4" + data_type: TYPE_UINT8 + dims: [ 4 ] + reshape { + shape: [ 4 ] + } } ] output [ { - name: "Softmax" + name: "Softmax11" data_type: TYPE_FP32 dims: [ -1, 10 ] }, { - name: "Softmax2" + name: "Softmax222" data_type: TYPE_UINT16 dims: [ 1 ] reshape {