Skip to content

Commit 789ce18

Browse files
author
gushiqiao
committed
Fix dp bugs
1 parent f63114c commit 789ce18

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

llmc/__main__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,9 @@ def main(config):
5454

5555
if 'eval' in config and 'pretrain' in config.eval.eval_pos:
5656
if config.eval.type == 'acc':
57-
acc_eval = AccuracyEval(eval_config)
58-
eval_list.append(acc_eval)
57+
for acc_eval in eval_list:
58+
acc = acc_eval.eval(model)
59+
logger.info(f'{config.eval.name} acc : {acc}')
5960
else:
6061
for ppl_eval in eval_list:
6162
ppl = ppl_eval.eval(model)
@@ -213,7 +214,6 @@ def main(config):
213214
)
214215
logger.info(f'opencompass_cmd : {opencompass_cmd}')
215216
os.system(opencompass_cmd)
216-
217217
dist.barrier()
218218

219219

0 commit comments

Comments
 (0)