Update config.yml
Browse files- config.yml +2 -2
config.yml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
bert_gen:
|
| 2 |
config_path: config.json
|
| 3 |
-
device:
|
| 4 |
num_processes: 2
|
| 5 |
use_multi_device: false
|
| 6 |
dataset_path: Data\model_name
|
|
@@ -27,7 +27,7 @@ server:
|
|
| 27 |
port: 5000
|
| 28 |
style_gen:
|
| 29 |
config_path: config.json
|
| 30 |
-
device:
|
| 31 |
num_processes: 4
|
| 32 |
train_ms:
|
| 33 |
config_path: config.json
|
|
|
|
| 1 |
bert_gen:
|
| 2 |
config_path: config.json
|
| 3 |
+
device: cpu
|
| 4 |
num_processes: 2
|
| 5 |
use_multi_device: false
|
| 6 |
dataset_path: Data\model_name
|
|
|
|
| 27 |
port: 5000
|
| 28 |
style_gen:
|
| 29 |
config_path: config.json
|
| 30 |
+
device: cpu
|
| 31 |
num_processes: 4
|
| 32 |
train_ms:
|
| 33 |
config_path: config.json
|