diff --git a/1/model.pt b/1/model.pt new file mode 100644 index 0000000..0e30d0b --- /dev/null +++ b/1/model.pt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e5c98a763cd04ae17252507a628fb7fc15343342f5cdcd457d3f35c99851f493 +size 5933728 diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..b7750f2 --- /dev/null +++ b/config.pbtxt @@ -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 + } +] \ No newline at end of file