diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-07 14:34:37 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-01-07 14:34:37 +0800 |
| commit | a27af5dfd58e7b48cf3bd063fa2b4b51ed1e0277 (patch) | |
| tree | 08e6b52d4e59506e6890bf67d1922abea97ce116 /utils | |
| parent | f1fe77c083f952e81cf80c0b44611fc6057a7882 (diff) | |
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
Diffstat (limited to 'utils')
| -rw-r--r-- | utils/configuration.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/utils/configuration.py b/utils/configuration.py index 3e98343..f3ae0b3 100644 --- a/utils/configuration.py +++ b/utils/configuration.py @@ -1,12 +1,10 @@ from typing import TypedDict, Optional, Union -import torch - from utils.dataset import ClipClasses, ClipConditions, ClipViews class SystemConfiguration(TypedDict): - device: torch.device + disable_acc: bool CUDA_VISIBLE_DEVICES: str save_dir: str |