fix import issues

Signed-off-by: zehao-intel <[email protected]>

examples/3.x_api/tensorflow/cv/resnet_v2_50/quantization/ptq/main.py

@@ -118,7 +118,6 @@ def run(self):
             q_model.save(args.output_graph)
 
         if args.benchmark:
-            dataloader = create_dataloader('tensorflow', dataloader_args)
             dataset = ImageRecordDataset(
                 root=args.dataset_location, 
                 transform=ComposeTransform(transform_list= [

examples/3.x_api/tensorflow/nlp/transformer_lt/quantization/ptq/main.py

@@ -143,9 +143,7 @@ def eval_func(infer_graph, iteration=-1):
         'model/Transformer/strided_slice_19:0')
 
     ds = Dataset(FLAGS.inputs_file, FLAGS.reference_file, FLAGS.vocab_file)
-    dataloader = DataLoader(framework='tensorflow', dataset=ds,
-                                batch_size=FLAGS.batch_size, collate_fn=collate_fn)
-
+    dataloader = BaseDataLoader(dataset=ds, batch_size=FLAGS.batch_size, collate_fn=collate_fn)
     config = tf.compat.v1.ConfigProto()
     config.use_per_session_threads = 1
     config.inter_op_parallelism_threads = 1

examples/3.x_api/tensorflow/object_detection/faster_rcnn_resnet50/quantization/ptq/main.py

@@ -16,13 +16,13 @@
 # limitations under the License.
 #
 #
+from __future__ import division
 
 import time
 
 import numpy as np
 import tensorflow as tf
 
-from __future__ import division
 from argparse import ArgumentParser
 from data_process import(
     COCOmAPv2,

examples/3.x_api/tensorflow/object_detection/mask_rcnn_inception_v2/quantization/ptq/main.py

@@ -16,13 +16,13 @@
 # limitations under the License.
 #
 #
+from __future__ import division
 
 import time
 
 import numpy as np
 import tensorflow as tf
 
-from __future__ import division
 from argparse import ArgumentParser
 from data_process import(
     COCOmAPv2,

examples/3.x_api/tensorflow/object_detection/ssd_mobilenet_v1/quantization/ptq/main.py

@@ -16,13 +16,13 @@
 # limitations under the License.
 #
 #
+from __future__ import division
 
 import time
 
 import numpy as np
 import tensorflow as tf
 
-from __future__ import division
 from argparse import ArgumentParser
 from data_process import(
     COCOmAPv2,
@@ -93,7 +93,7 @@ def eval_func(dataloader):
 
     eval_dataset = COCORecordDataset(root=args.dataset_location, filter=None, \
             transform=ComposeTransform(transform_list=[ResizeTFTransform(size=300)]))
-    eval_dataloader=TFDataLoader(framework='tensorflow', dataset=eval_dataset, batch_size=args.batch_size)
+    eval_dataloader=TFDataLoader(dataset=eval_dataset, batch_size=args.batch_size)
     latency = eval_func(eval_dataloader)
     if args.benchmark and args.mode == 'performance':
         print("Batch size = {}".format(args.batch_size))
@@ -105,7 +105,7 @@ def eval_func(dataloader):
 def main(_):
     calib_dataset = COCORecordDataset(root=args.dataset_location, filter=None, \
             transform=ComposeTransform(transform_list=[ResizeTFTransform(size=300)]))
-    calib_dataloader = TFDataLoader(framework='tensorflow', dataset=calib_dataset, batch_size=args.batch_size)
+    calib_dataloader = TFDataLoader(dataset=calib_dataset, batch_size=args.batch_size)
 
     if args.tune:
         from neural_compressor.tensorflow import StaticQuantConfig, quantize_model, Model

examples/3.x_api/tensorflow/recommendation/wide_deep_large_ds/quantization/ptq/main.py

@@ -18,6 +18,10 @@
 
 #
 
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
 import sys
 import os
 import numpy as np
@@ -30,10 +34,6 @@
 
 import tensorflow as tf
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 from tensorflow.python.framework import ops
 from tensorflow.core.framework import graph_pb2
 from google.protobuf import text_format