diff --git a/scripts/calibrator.py b/scripts/calibrator.py index b14b69bf4..bab3f4363 100644 --- a/scripts/calibrator.py +++ b/scripts/calibrator.py @@ -577,7 +577,7 @@ def generate_dummy_model(model_path, dummy): if params.calibration_algos != 'DIRECT' and params.calibration_algos != "KL" and params.calibration_algos != "MAXP": user_calibration_algos = 'DIRECT' else: - user_calibration_alogs = params.calibration_algos + user_calibration_algos = params.calibration_algos if params.conv_algo != False and params.conv_algo != True: user_conv_algo = False @@ -627,7 +627,7 @@ def generate_dummy_model(model_path, dummy): quantized_prototxt = model.rsplit('.')[0] + '_quantized.prototxt' print 'Sampling...' - (top_blobs_map, bottom_blobs_map) = generate_sample(model, user_input_weights, quantized_prototxt, user_scaling_mode, user_calibration_alogs, user_conv_algo, user_sampling_iteration, user_enable_1st_conv) + (top_blobs_map, bottom_blobs_map) = generate_sample(model, user_input_weights, quantized_prototxt, user_scaling_mode, user_calibration_algos, user_conv_algo, user_sampling_iteration, user_enable_1st_conv) print 'Sampling done' top_1 = None if not params.quantize_model: