run on cpu
Browse files
test.yml
CHANGED
|
@@ -36,7 +36,7 @@ post_cbhg_use_batch_norm: true
|
|
| 36 |
|
| 37 |
use_mixed_precision: false
|
| 38 |
optimizer_type: Adam
|
| 39 |
-
device:
|
| 40 |
|
| 41 |
# LOGGING
|
| 42 |
evaluate_frequency: 50000000
|
|
|
|
| 36 |
|
| 37 |
use_mixed_precision: false
|
| 38 |
optimizer_type: Adam
|
| 39 |
+
device: cpu
|
| 40 |
|
| 41 |
# LOGGING
|
| 42 |
evaluate_frequency: 50000000
|