from mmengine.utils import get_git_hash | |
from mmengine.utils.dl_utils import collect_env as collect_base_env | |
import opencompass | |
def collect_env(): | |
"""Collect the information of the running environments.""" | |
env_info = collect_base_env() | |
env_info['opencompass'] = opencompass.__version__ + '+' + get_git_hash( | |
)[:7] | |
return env_info | |