Commit
•
c7fa5e9
1
Parent(s):
c02e05d
Upload 2 files (#2)
Browse files- Upload 2 files (23e47ccecd903dfe496d2dba64353663bfb92f80)
Co-authored-by: DehuaTang <DehuaTang@users.noreply.huggingface.co>
- eval_onnx.py +1 -1
- mnasnet_b1_int.onnx +2 -2
eval_onnx.py
CHANGED
@@ -144,7 +144,7 @@ def val_imagenet():
|
|
144 |
val_loader = tqdm(val_loader, file=sys.stdout)
|
145 |
with torch.no_grad():
|
146 |
for batch_idx, (images, targets) in enumerate(val_loader):
|
147 |
-
inputs, targets = images.numpy(), targets
|
148 |
ort_inputs = {ort_session.get_inputs()[0].name: inputs}
|
149 |
|
150 |
outputs = ort_session.run(None, ort_inputs)
|
|
|
144 |
val_loader = tqdm(val_loader, file=sys.stdout)
|
145 |
with torch.no_grad():
|
146 |
for batch_idx, (images, targets) in enumerate(val_loader):
|
147 |
+
inputs, targets = images.permute([0,2,3,1]).numpy(), targets
|
148 |
ort_inputs = {ort_session.get_inputs()[0].name: inputs}
|
149 |
|
150 |
outputs = ort_session.run(None, ort_inputs)
|
mnasnet_b1_int.onnx
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:573e80009e1d0c21858d507ee6702ea73d092f9bf8be09ac7d901f0259e4bc1f
|
3 |
+
size 17571818
|