From a27af5dfd58e7b48cf3bd063fa2b4b51ed1e0277 Mon Sep 17 00:00:00 2001
From: Jordan Gong <jordan.gong@protonmail.com>
Date: Thu, 7 Jan 2021 14:34:37 +0800
Subject: Change device config and add enable multi-GPU computing

1. Add `disable_acc` switch for disabling accelerator. When it is off, system will automatically choosing accelerator.
2. Enable multi-GPU training using torch.nn.DataParallel
---
 config.py | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

(limited to 'config.py')

diff --git a/config.py b/config.py
index ad737e8..47ded38 100644
--- a/config.py
+++ b/config.py
@@ -1,12 +1,10 @@
-import torch
-
 from utils.configuration import Configuration
 
 config: Configuration = {
     'system': {
-        # Device(s) used in training and testing (CPU or CUDA)
-        'device': torch.device('cuda'),
-        # GPU(s) used in training or testing, if CUDA enabled
+        # Disable accelerator
+        'disable_acc': False,
+        # GPU(s) used in training or testing if available
         'CUDA_VISIBLE_DEVICES': '0',
         # Directory used in training or testing for temporary storage
         'save_dir': 'runs',
-- 
cgit v1.2.3