|
424 | 424 | " raw_samples=RAW_SAMPLES, # used for intialization heuristic\n",
|
425 | 425 | " options={\n",
|
426 | 426 | " \"batch_limit\": 5,\n",
|
427 |
| - " \"maxiter\": 20 if SMOKE_TEST else 200,\n", |
| 427 | + " \"maxiter\": 3 if SMOKE_TEST else 200,\n", |
428 | 428 | " \"nonnegative\": True,\n",
|
429 | 429 | " },\n",
|
430 | 430 | " sequential=True,\n",
|
|
479 | 479 | "from torch import Tensor\n",
|
480 | 480 | "\n",
|
481 | 481 | "NUM_INNER_MC_SAMPLES = 2 if SMOKE_TEST else 32\n",
|
482 |
| - "NUM_PARETO = 3 if SMOKE_TEST else 10\n", |
| 482 | + "NUM_PARETO = 1 if SMOKE_TEST else 10\n", |
483 | 483 | "NUM_FANTASIES = 2 if SMOKE_TEST else 8\n",
|
484 | 484 | "\n",
|
485 | 485 | "\n",
|
|
514 | 514 | " bounds=bounds[:, non_fidelity_dims],\n",
|
515 | 515 | " q=NUM_PARETO,\n",
|
516 | 516 | " num_restarts=1,\n",
|
517 |
| - " raw_samples=1024,\n", |
| 517 | + " raw_samples=2 * RAW_SAMPLES,\n", |
518 | 518 | " return_best_only=True,\n",
|
519 |
| - " options={\"nonnegative\": True},\n", |
| 519 | + " options={\n", |
| 520 | + " \"nonnegative\": True,\n", |
| 521 | + " \"maxiter\": 3 if SMOKE_TEST else 200,\n", |
| 522 | + " },\n", |
520 | 523 | " )\n",
|
521 | 524 | " return current_value\n",
|
522 | 525 | "\n",
|
|
572 | 575 | " q=BATCH_SIZE,\n",
|
573 | 576 | " num_restarts=1,\n",
|
574 | 577 | " raw_samples=RAW_SAMPLES, # used for intialization heuristic\n",
|
575 |
| - " options={\"batch_limit\": 5},\n", |
| 578 | + " options={\n", |
| 579 | + " \"batch_limit\": 5,\n", |
| 580 | + " \"maxiter\": 3 if SMOKE_TEST else 200,\n", |
| 581 | + " },\n", |
576 | 582 | " )\n",
|
577 | 583 | " # if the AF val is 0, set the fidelity parameter to zero\n",
|
578 | 584 | " if vals.item() == 0.0:\n",
|
|
0 commit comments