Upload 2 files

eval_onnx.py

@@ -144,7 +144,7 @@ def val_imagenet():
   val_loader = tqdm(val_loader, file=sys.stdout)
   with torch.no_grad():
     for batch_idx, (images, targets) in enumerate(val_loader):
-      inputs, targets = images.numpy(), targets
+      inputs, targets = images.permute([0,2,3,1]).numpy(), targets
       ort_inputs = {ort_session.get_inputs()[0].name: inputs}
 
       outputs = ort_session.run(None, ort_inputs)

mnasnet_b1_int.onnx

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a4784fa94c18ad2cb8c91081663cbdc825ba7bc61d1d913abd50d0fe0ff84949
-size 17571696
+oid sha256:573e80009e1d0c21858d507ee6702ea73d092f9bf8be09ac7d901f0259e4bc1f
+size 17571818