Update eval_onnx.py

eval_onnx.py

@@ -21,8 +21,6 @@ parser.add_argument(
     "--data_dir",
     default="/workspace/dataset/imagenet",
     help="Directory of dataset")
-parser.add_argument(
-    "--batch_size", default=1, type=int, help="Evaluation batch size")
 parser.add_argument(
     "--ipu",
     action="store_true",
@@ -87,7 +85,7 @@ def accuracy(output: torch.Tensor,
     return res
 
 def prepare_data_loader(data_dir: str,
-                        batch_size: int = 100,
+                        batch_size: int = 1,
                         workers: int = 8) -> torch.utils.data.DataLoader:
   """Returns a validation data loader of ImageNet by given `data_dir`.
   Args:
@@ -132,7 +130,7 @@ def val_imagenet():
   ort_session = onnxruntime.InferenceSession(
       args.onnx_model, providers=providers, provider_options=provider_options)
 
-  val_loader = prepare_data_loader(args.data_dir, args.batch_size)
+  val_loader = prepare_data_loader(args.data_dir)
 
   top1 = AverageMeter('Acc@1', ':6.2f')
   top5 = AverageMeter('Acc@5', ':6.2f')
@@ -158,4 +156,4 @@ def val_imagenet():
   return top1.avg, top5.avg
 
 if __name__ == '__main__':
-  val_imagenet()
+  val_imagenet()