Update config.pbtxt
This commit is contained in:
parent
7fd5a661da
commit
5f8e92a371
13
config.pbtxt
13
config.pbtxt
@ -4,30 +4,21 @@ platform: "onnxruntime_onnx"
|
|||||||
max_batch_size: 0
|
max_batch_size: 0
|
||||||
|
|
||||||
input [
|
input [
|
||||||
|
|
||||||
{
|
{
|
||||||
name: "images"
|
name: "images"
|
||||||
data_type: "TYPE_FP32"
|
data_type: TYPE_FP32
|
||||||
dims: [ 1, 3, 640, 640 ]
|
dims: [ 1, 3, 640, 640 ]
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
]
|
]
|
||||||
|
|
||||||
output [
|
output [
|
||||||
|
|
||||||
{
|
{
|
||||||
name: "output0"
|
name: "output0"
|
||||||
data_type: "TYPE_FP32"
|
data_type: TYPE_FP32
|
||||||
dims: [ 1, 84, 8400 ]
|
dims: [ 1, 84, 8400 ]
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
instance_group [
|
instance_group [
|
||||||
{
|
{
|
||||||
kind: KIND_AUTO
|
kind: KIND_AUTO
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user