mirror of
https://github.com/open-mmlab/mmpretrain.git
synced 2025-06-03 14:59:18 +08:00
b7d0d521eb
* [Refactor] Refactor the rest data transforms and fix some docstring. * Use `mmengine` instead of `mmcv` * Refactor KFold dataset tools * Fix docstring according to comments
13 lines
379 B
Python
13 lines
379 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from mmengine.utils import collect_env as collect_base_env
|
|
from mmengine.utils import get_git_hash
|
|
|
|
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]
|
|
return env_info
|