DeepSeek/DeepSeek-Math
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Math synced 2024-11-21 19:17:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

update submit_eval_jobs

Browse Source
This commit is contained in:
ZhihongShao 2024-02-09 15:23:28 +08:00
parent 83d3c4cc6a
commit db877abb91
4 changed files with 32 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
evaluation/infer/run_cot_eval.py
View File

@ -1,5 +1,10 @@
import argparse
import os
import sys
import_path = os.path.abspath(__file__)
for _ in range(2):
import_path = os.path.dirname(import_path)
sys.path.append(import_path)
from tqdm import tqdm
import json

6
evaluation/infer/run_pal_eval.py
View File

@ -1,5 +1,11 @@
import argparse
import os
import sys
import_path = os.path.abspath(__file__)
for _ in range(2):
import_path = os.path.dirname(import_path)
sys.path.append(import_path)
import json
from copy import deepcopy
from functools import partial

6
evaluation/infer/run_tool_integrated_eval.py
View File

@ -1,5 +1,11 @@
import argparse
import os
import sys
import_path = os.path.abspath(__file__)
for _ in range(2):
import_path = os.path.dirname(import_path)
sys.path.append(import_path)
from tqdm import tqdm
import regex
import json

4
evaluation/submit_eval_jobs.py
View File

@ -46,7 +46,6 @@ def main():
args = parser.parse_args()
conf = base_conf # TODO: your conf here
for rank in range(args.n_gpus):
cmd = "python run_subset_parallel.py"
for key, val in conf.items():
if key == 'expname':
@ -58,7 +57,8 @@ def main():
cmd += f" --test-conf {conf['test-conf']}"
cmd += f" --n-repeats {args.n_repeats}"
cmd += f" --temperature {args.temperature}"
cmd += f" --rank {rank} &"
cmd += f" --ngpus {args.n_gpus}"
cmd += f" --rank {0} &"
print(cmd, flush=True)
os.system(cmd)