diff --git a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py index 812c276798a6cfd54bfeb3bd05b7fa01ad404758..edb8d4af29ddb81aad51176590864ad0f9bdf999 100644 --- a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py +++ b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py @@ -960,7 +960,7 @@ class TensorRtTranslator: layer_knobs_str += self.knobs_str[:-3] layer_knobs_str += " \n\n }, \n\n" - layer_knobs_str += "\"baseline_knob\" : " + baseline_knobs + ", \n" + layer_knobs_str += "\"baseline_knob\" : " + fp32_baseline + ", \n" labels_path = weights_dir + "/tune_labels.bin" layer_knobs_str += "\"tune_labels_path\" : \"" + labels_path + "\", \n" layer_knobs_str += "\"conf_path\" : \"tuner_confs.txt\", \n" diff --git a/hpvm/projects/keras/keras_frontend/knobs.py b/hpvm/projects/keras/keras_frontend/knobs.py index 388cbd28f804f9b92a8141b9273b226495bc9acc..fb0c029478274aafc2f687b3ea676a1b1306105e 100644 --- a/hpvm/projects/keras/keras_frontend/knobs.py +++ b/hpvm/projects/keras/keras_frontend/knobs.py @@ -317,6 +317,7 @@ for i in range(261, 270): conv_knobs += ", " -baseline_knobs = "\"12\"" +fp32_baseline = "\"11\"" +baseline_knobs = "\"11\", \"12\""