Upload config.pbtxt
This commit is contained in:
parent
066aed90c9
commit
1d951bc9ca
17
config.pbtxt
Normal file
17
config.pbtxt
Normal file
@ -0,0 +1,17 @@
|
|||||||
|
name: "pytorch-fashion-minst"
|
||||||
|
platform: "pytorch_libtorch"
|
||||||
|
max_batch_size : 0
|
||||||
|
input [
|
||||||
|
{
|
||||||
|
name: "input__0"
|
||||||
|
data_type: TYPE_FP32
|
||||||
|
dims: [ 1, 1, 28, 28 ]
|
||||||
|
}
|
||||||
|
]
|
||||||
|
output [
|
||||||
|
{
|
||||||
|
name: "output__0"
|
||||||
|
data_type: TYPE_FP32
|
||||||
|
dims: [ 10 ]
|
||||||
|
}
|
||||||
|
]
|
||||||
Loading…
Reference in New Issue
Block a user