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]>

Author: TedThemistokleous

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"