Update resnet50 test (#398)

* Add additional flags for resnet50 int8 run

- Allow for mixed precsion via fp16 flag
-Allow for variable batch
-Allow for variable calibrate data size

* Fix datatype for os.environ vars

---------

Co-authored-by: Ted Themistokleous <[email protected]>

quantization/image_classification/migraphx/resnet50/e2e_migraphx_resnet_example.py

@@ -398,9 +398,9 @@ def get_dataset_size(dataset_path, calibration_dataset_size):
  print("Write complete")
 
  if flags.fp16:
- os.environ["ORT_MIGRAPHX_FP16_ENABLE"] = 1
+ os.environ["ORT_MIGRAPHX_FP16_ENABLE"] = "1"
  else:
- os.environ["ORT_MIGRAPHX_FP16_ENABLE"] = 0
+ os.environ["ORT_MIGRAPHX_FP16_ENABLE"] = "0"
 
  # Run prediction in MIGraphX EP138G
  data_reader = ImageNetDataReader(ilsvrc2012_dataset_path,
@@ -422,5 +422,6 @@ def get_dataset_size(dataset_path, calibration_dataset_size):
  evaluator.evaluate(result)
 
  #Set OS flags to off to ensure we don't interfere with other test runs
- os.environ["ORT_MIGRAPHX_FP16_ENABLE"] = 0
- os.environ["ORT_MIGRAPHX_INT8_ENABLE"] = 0
+
+ os.environ["ORT_MIGRAPHX_FP16_ENABLE"] = "0"
+ os.environ["ORT_MIGRAPHX_INT8_ENABLE"] = "0"