DehuaTang commited on
Commit
5c97625
1 Parent(s): 221c36e

Upload 2 files

Browse files
Files changed (2) hide show
  1. ese_vovnet39b_int.onnx +2 -2
  2. eval_onnx.py +1 -1
ese_vovnet39b_int.onnx CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e54c608973ec2515497ebac87955f35900334235c6c235b2db73e0d154cbe8df
3
- size 102032170
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e056581c127708ee9751d6402fe7baa9c9fced577a240054e1c982d8a1df6856
3
+ size 102032283
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)