Spaces:
Runtime error
Runtime error
File size: 561 Bytes
3b96cb1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# Copyright (c) OpenMMLab. All rights reserved.
import mmcv
from mmengine.utils import get_git_hash
from mmengine.utils.dl_utils import collect_env as collect_base_env
import mmpretrain
def collect_env(with_torch_comiling_info=False):
"""Collect the information of the running environments."""
env_info = collect_base_env()
env_info['MMCV'] = mmcv.__version__
if not with_torch_comiling_info:
env_info.pop('PyTorch compiling details')
env_info['MMPreTrain'] = mmpretrain.__version__ + '+' + get_git_hash()[:7]
return env_info
|