fix reshape issue for bayesian strategy (#1042)

Signed-off-by: Kaihui-intel <[email protected]>
intel · Jun 29, 2023 · 77cb836 · 77cb836
1 parent 501440a
commit 77cb836
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_compressor/strategy/bayesian.py b/neural_compressor/strategy/bayesian.py
@@ -172,7 +172,7 @@ def acq_max(ac, gp, y_max, bounds, random_seed, n_warmup=10000, n_iter=10):
     for x_try in x_seeds:
         # Find the minimum of minus the acquisition function
         res = minimize(lambda x: -ac(x.reshape(1, -1), gp=gp, y_max=y_max),
-                       x_try.reshape(1, -1),
+                       x_try.flatten(),
                        bounds=bounds,
                        method="L-BFGS-B")