16 lines
484 B
Python
16 lines
484 B
Python
from custom_mmpkg.custom_mmcv.utils import collect_env as collect_base_env
|
|
from custom_mmpkg.custom_mmcv.utils import get_git_hash
|
|
|
|
import motiondiff_modules.mogen as mogen
|
|
|
|
|
|
def collect_env():
|
|
"""Collect the information of the running environments."""
|
|
env_info = collect_base_env()
|
|
env_info['mogen'] = mogen.__version__ + '+' + get_git_hash()[:7]
|
|
return env_info
|
|
|
|
|
|
if __name__ == '__main__':
|
|
for name, val in collect_env().items():
|
|
print(f'{name}: {val}') |