mmclassification/mmcls/utils/collect_env.py

13 lines
388 B
Python
Raw Normal View History

# Copyright (c) OpenMMLab. All rights reserved.
2022-07-14 14:05:52 +08:00
from mmengine.utils import get_git_hash
from mmengine.utils.dl_utils import collect_env as collect_base_env
2020-05-21 21:21:43 +08:00
import mmcls
def collect_env():
"""Collect the information of the running environments."""
env_info = collect_base_env()
env_info['MMClassification'] = mmcls.__version__ + '+' + get_git_hash()[:7]
2020-05-21 21:21:43 +08:00
return env_info