[python]: update yamrt

wlt-cortex · wlt-cortex · commit c2aa45dd5b8a · 2022-11-17T17:33:49.000+08:00
diff --git a/.github/workflows/ccpp.yml b/.github/workflows/ccpp.yml
diff --git a/python/mrt/yamrt/quant/quantize.py b/python/mrt/yamrt/quant/quantize.py
@@ -106,7 +106,7 @@ def quantize_static(model_input,
 
     model = load_model(Path(model_input), optimize_model)
 
-    calibrator = get_calibrator(model, op_types_to_quantize, calibrate_method=calibrate_method)
+    calibrator = get_calibrator(model, op_types_to_quantize, method=calibrate_method)
     calibrator.collect_data(calibration_data_reader)
     tensors_range = calibrator.compute_range()
 
diff --git a/tests/yamrt/infer.py b/tests/yamrt/infer.py
@@ -46,7 +46,7 @@ def softmax(x):
         for i in range(data.shape[0]):
             norm_data[i,:,:] = (data[i,:,:]/255 - mean_vec[i]) / stddev_vec[i]
         norm_data = norm_data.reshape(1, 3, 224, 224).astype('float32')
-        
+
         result = session.run([output_name],{input_name:norm_data})
         res = softmax(np.array(result)).tolist()
         idx = np.argmax(res)
