Unverified Commit 889a6b26 authored by Hubert's avatar Hubert Committed by GitHub
Browse files

[Fix] fix log re-direct (#564)

parent cf5a6d1a
from mmengine.config import read_base from mmengine.config import read_base
from opencompass.models.minimax import MiniMax from opencompass.models import MiniMax
from opencompass.partitioners import NaivePartitioner from opencompass.partitioners import NaivePartitioner
from opencompass.runners import LocalRunner from opencompass.runners import LocalRunner
from opencompass.runners.local_api import LocalAPIRunner from opencompass.runners.local_api import LocalAPIRunner
......
...@@ -4,6 +4,7 @@ import os.path as osp ...@@ -4,6 +4,7 @@ import os.path as osp
import subprocess import subprocess
import sys import sys
import time import time
import traceback
from multiprocessing import Manager, Pool from multiprocessing import Manager, Pool
from multiprocessing.managers import SyncManager from multiprocessing.managers import SyncManager
from typing import Any, Dict, List, Tuple from typing import Any, Dict, List, Tuple
...@@ -122,9 +123,13 @@ def launch(task: BaseTask, tokens: SyncManager.Semaphore): ...@@ -122,9 +123,13 @@ def launch(task: BaseTask, tokens: SyncManager.Semaphore):
end_time = time.time() end_time = time.time()
logger.info(f'time elapsed: {end_time - start_time:.2f}s') logger.info(f'time elapsed: {end_time - start_time:.2f}s')
except Exception: 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}') logger.warning(f'task {task_name} fail, see\n{out_path}')
returncode = 1 returncode = 1
finally: else:
# reset stdout and stderr # reset stdout and stderr
reset_std() reset_std()
return task_name, returncode return task_name, returncode
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment