summaryrefslogtreecommitdiff
path: root/utils/misc.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2021-02-19 22:45:29 +0800
committerJordan Gong <jordan.gong@protonmail.com>2021-02-19 22:45:29 +0800
commitd098956ab767464909b606afa6a5f39f756d90a6 (patch)
tree33617f30b1b453f841f0841b68a2d5e3b0e1dbdb /utils/misc.py
parent4049566103a00aa6d5a0b1f73569bdc5435714ca (diff)
parentc6bff0a8350da4f4d42b6d17052c8fae37633237 (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.py5
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: