Update config.yaml
Browse files- config.yaml +1 -1
config.yaml
CHANGED
|
@@ -99,6 +99,6 @@ SYSTEM:
|
|
| 99 |
use_seed: false
|
| 100 |
seed: 42
|
| 101 |
GPU: 4
|
| 102 |
-
device:
|
| 103 |
gpu_id: 0
|
| 104 |
|
|
|
|
| 99 |
use_seed: false
|
| 100 |
seed: 42
|
| 101 |
GPU: 4
|
| 102 |
+
device: auto # auto | cuda | cpu
|
| 103 |
gpu_id: 0
|
| 104 |
|