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 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
|
|||||||
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):
|
|||||||
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
|
||||||
|
Loading…
Reference in New Issue
Block a user