Spaces:
Sleeping
Sleeping
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
|