mirror of
https://github.com/open-compass/opencompass.git
synced 2025-05-30 16:03:24 +08:00
[Fix] fix log re-direct (#564)
This commit is contained in:
parent
cf5a6d1ab7
commit
889a6b26ae
@ -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
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user