Speed up Constrained MOBO tutorial in smoke-test mode (#2464)

Summary: Pull Request resolved: #2464 This PR reduces the number of BayesOpt iterations to 1 in smoke-test mode. Reviewed By: Balandat Differential Revision: D61085444 fbshipit-source-id: a6096b29f1a3edb9e7498259c92ccb8df32ddc0d
pytorch · Aug 12, 2024 · 3f5fcd6 · 3f5fcd6
1 parent bcdea09
commit 3f5fcd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/constrained_multi_objective_bo.ipynb b/tutorials/constrained_multi_objective_bo.ipynb
@@ -393,7 +393,7 @@
         "warnings.filterwarnings(\"ignore\", category=BadInitialCandidatesWarning)\n",
         "warnings.filterwarnings(\"ignore\", category=RuntimeWarning)\n",
         "\n",
-        "N_BATCH = 20 if not SMOKE_TEST else 5\n",
+        "N_BATCH = 20 if not SMOKE_TEST else 1\n",
         "MC_SAMPLES = 128 if not SMOKE_TEST else 16\n",
         "verbose = True\n",
         "\n",