diff options
author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-19 22:45:29 +0800 |
---|---|---|
committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-19 22:45:29 +0800 |
commit | d098956ab767464909b606afa6a5f39f756d90a6 (patch) | |
tree | 33617f30b1b453f841f0841b68a2d5e3b0e1dbdb /utils/misc.py | |
parent | 4049566103a00aa6d5a0b1f73569bdc5435714ca (diff) | |
parent | c6bff0a8350da4f4d42b6d17052c8fae37633237 (diff) |
Merge branch 'python3.7' into disentangling_only_py3.7disentangling_only_py3.7
# Conflicts:
# models/model.py
# utils/configuration.py
Diffstat (limited to 'utils/misc.py')
-rw-r--r-- | utils/misc.py | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/utils/misc.py b/utils/misc.py index b850830..c047618 100644 --- a/utils/misc.py +++ b/utils/misc.py @@ -1,9 +1,8 @@ import os +from typing import Dict -from utils.configuration import SystemConfiguration - -def set_visible_cuda(config: SystemConfiguration): +def set_visible_cuda(config: Dict): """Set environment variable CUDA device(s)""" CUDA_VISIBLE_DEVICES = config.get('CUDA_VISIBLE_DEVICES', None) if CUDA_VISIBLE_DEVICES: |