Fixed DEFAULT_LABEL issue

apache · Feb 4, 2020 · fbebf35 · fbebf35
1 parent 20efc68
commit fbebf35
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/benchmark/opperf/rules/default_params.py b/benchmark/opperf/rules/default_params.py
@@ -77,7 +77,7 @@
 DEFAULT_BETA = [(3,)]
 DEFAULT_MOVING_MEAN = [(3,)]
 DEFAULT_MOVING_VAR = [(3,)]
-DEFAULT_LABEL = [(32, 3, 256, 256), (32, 3, 10000, 10)]
+DEFAULT_LABEL_REG = [(32, 3, 256, 256), (32, 3, 10000, 10)]
 DEFAULT_GRAD_SCALE = [.5]
 DEFAULT_NORMALIZATION = ["batch"]
 DEFAULT_MARGIN = [.5]
@@ -281,9 +281,9 @@
                    "moving_mean": DEFAULT_MOVING_MEAN,
                    "moving_var": DEFAULT_MOVING_VAR,
                    "label_softmaxoutput": DEFAULT_LABEL_SM,
-                   "label_maeregressionoutput": DEFAULT_LABEL,
-                   "label_logisticregressionoutput": DEFAULT_LABEL,
-                   "label_linearregressionoutput": DEFAULT_LABEL,
+                   "label_maeregressionoutput": DEFAULT_LABEL_REG,
+                   "label_logisticregressionoutput": DEFAULT_LABEL_REG,
+                   "label_linearregressionoutput": DEFAULT_LABEL_REG,
                    "label_svmoutput": DEFAULT_LABEL_SVM,
                    "grad_scale": DEFAULT_GRAD_SCALE,
                    "normalization": DEFAULT_NORMALIZATION,