groupuser/pytorch_fashionmnist
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add pytorch mnist

Browse Source
This commit is contained in:
Jaeyoon Jeong 2024-09-05 18:12:09 +09:00
parent 095be78fdb
commit 89931b8280
2 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
1/model.pt (Stored with Git LFS) Normal file
View File

Binary file not shown.

22
config.pbtxt Normal file
View File

@ -0,0 +1,22 @@
name: "pytorch_fashionmnist"
platform: "pytorch_libtorch"
max_batch_size: 64
input [
{
name: "input__0"
data_type: TYPE_FP32
dims: [ 1, 28, 28 ]
}
]
output [
{
name: "output__0"
data_type: TYPE_FP32
dims: [ 10 ]
}
]
instance_group [
{
kind: KIND_GPU
}
]