github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix default values of data item fields

pull/695/head
wangzy 2024-02-05 14:22:52 +08:00
parent b81132283d
commit 4e589c9cbf
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
agent/streaming_inference.py
View File

@ -518,7 +518,7 @@ def predict(args):
d['idx'] = k d['idx'] = k
d['query'] = d['question'].replace('$', '') d['query'] = d['question'].replace('$', '')
d['gt'] = re.search('The answer is (.+)', d['answer'])[1] d['gt'] = re.search('The answer is (.+)', d['answer'])[1]
d['pred'], d['steps'], d['error'] = None, [], [] d['pred'], d['steps'], d['error'] = [], [], None
return d return d
dataset = load_dataset('gsm8k', 'main', dataset = load_dataset('gsm8k', 'main',