|
2 | 2 | This script runs the GPT_3 experiments and prints the results to stdout. |
3 | 3 | It uses cacheing mechanisms so that if run twice with the same parameters, it will give exactly the same |
4 | 4 | results and will not query the GPT3 API again and will not judge the resulting solutions again. Hence, the first |
5 | | -time you run it, it will be slow, but you can subsequently run it again and it will be fast. |
6 | | -
|
7 | | -It was run with sys.version = '3.6.9 (default, Jan 26 2021, 15:33:00) \n[GCC 8.4.0]' |
8 | | -python (3.6.9) |
9 | | -astor (0.8.1) |
10 | | -openai (0.6.3) |
11 | | -transformers (2.11.0) |
12 | | -Pebble (4.6.1) |
| 5 | +time you run it, it will be slow, but you can subsequently run it again and it will be fast. It will run the |
| 6 | +experiment three times, with different seeds to get different results. |
13 | 7 |
|
14 | 8 | """ |
15 | 9 |
|
@@ -120,11 +114,11 @@ def f5(n: int): |
120 | 114 |
|
121 | 115 |
|
122 | 116 | def run(seed=0): |
123 | | - sols = [lm_solve.prompt_experiment(**PARAMS, prefix=""), |
124 | | - lm_solve.prompt_experiment(**PARAMS, prefix=PREFIX), |
125 | | - lm_solve.prompt_experiment(**PARAMS, prefix=PREFIX_DOCSTR, add_docstring=True)] |
| 117 | + sols = [lm_solve.prompt_experiment(**PARAMS, prefix="", seed=seed), |
| 118 | + lm_solve.prompt_experiment(**PARAMS, prefix=PREFIX, seed=seed), |
| 119 | + lm_solve.prompt_experiment(**PARAMS, prefix=PREFIX_DOCSTR, add_docstring=True, seed=seed)] |
126 | 120 | problems_solved = [sorted([i for i, (f, gs) in enumerate(s) if gs]) for s in sols] |
127 | | - bootstrap = lm_solve.bootstrap(**BOOTSTRAP_PARAMS) |
| 121 | + bootstrap = lm_solve.bootstrap(**BOOTSTRAP_PARAMS, seed=seed) |
128 | 122 | print(f"run={seed} ALL DONE!\n\n") |
129 | 123 | print(f"run={seed} RESULTS " + "=" * 50) |
130 | 124 | print() |
|
0 commit comments