diff options
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/configuration.py | 0 | ||||
| -rw-r--r-- | utils/misc.py | 9 |
2 files changed, 0 insertions, 9 deletions
diff --git a/utils/configuration.py b/utils/configuration.py deleted file mode 100644 index e69de29..0000000 --- a/utils/configuration.py +++ /dev/null diff --git a/utils/misc.py b/utils/misc.py deleted file mode 100644 index c047618..0000000 --- a/utils/misc.py +++ /dev/null @@ -1,9 +0,0 @@ -import os -from typing import Dict - - -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: - os.environ['CUDA_VISIBLE_DEVICES'] = CUDA_VISIBLE_DEVICES |