diff --git a/configs/eval_minimax.py b/configs/eval_minimax.py index 6f88e6e3..54d0da36 100644 --- a/configs/eval_minimax.py +++ b/configs/eval_minimax.py @@ -1,5 +1,5 @@ from mmengine.config import read_base -from opencompass.models.minimax import MiniMax +from opencompass.models import MiniMax from opencompass.partitioners import NaivePartitioner from opencompass.runners import LocalRunner from opencompass.runners.local_api import LocalAPIRunner diff --git a/opencompass/runners/local_api.py b/opencompass/runners/local_api.py index c5346c24..04f78cf9 100644 --- a/opencompass/runners/local_api.py +++ b/opencompass/runners/local_api.py @@ -4,6 +4,7 @@ import os.path as osp import subprocess import sys import time +import traceback from multiprocessing import Manager, Pool from multiprocessing.managers import SyncManager from typing import Any, Dict, List, Tuple @@ -122,9 +123,13 @@ def launch(task: BaseTask, tokens: SyncManager.Semaphore): end_time = time.time() logger.info(f'time elapsed: {end_time - start_time:.2f}s') except Exception: + # print trace back in target file + traceback.print_exc() + # reset stdout and stderr + reset_std() logger.warning(f'task {task_name} fail, see\n{out_path}') returncode = 1 - finally: + else: # reset stdout and stderr reset_std() return task_name, returncode