Update scripts/run_pipeline.py
Browse files- scripts/run_pipeline.py +2 -1
scripts/run_pipeline.py
CHANGED
|
@@ -62,10 +62,11 @@ def main() -> None:
|
|
| 62 |
parser.add_argument("--no-eval", action="store_true", help="Skip benchmark evaluation")
|
| 63 |
parser.add_argument("--workers", type=int, default=4, help="Number of parallel workers")
|
| 64 |
parser.add_argument("--scenarios", type=int, default=None, help="Override number of scenarios")
|
|
|
|
| 65 |
args = parser.parse_args()
|
| 66 |
|
| 67 |
n_scenarios = args.scenarios or (50 if args.quick else 1000)
|
| 68 |
-
n_eval_seeds = 20 if args.quick else 1000
|
| 69 |
n_workers = args.workers
|
| 70 |
|
| 71 |
t_start = time.time()
|
|
|
|
| 62 |
parser.add_argument("--no-eval", action="store_true", help="Skip benchmark evaluation")
|
| 63 |
parser.add_argument("--workers", type=int, default=4, help="Number of parallel workers")
|
| 64 |
parser.add_argument("--scenarios", type=int, default=None, help="Override number of scenarios")
|
| 65 |
+
parser.add_argument("--eval-seeds", type=int, default=None, help="Override number of evaluation seeds")
|
| 66 |
args = parser.parse_args()
|
| 67 |
|
| 68 |
n_scenarios = args.scenarios or (50 if args.quick else 1000)
|
| 69 |
+
n_eval_seeds = args.eval_seeds or (20 if args.quick else 1000)
|
| 70 |
n_workers = args.workers
|
| 71 |
|
| 72 |
t_start = time.time()
|