Create New File
This commit is contained in:
parent
2cade0cd34
commit
1c086103fb
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