Zaid commited on
Commit
66368bd
1 Parent(s): 728e2c0

run on cpu

Browse files
Files changed (1) hide show
  1. test.yml +1 -1
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: cuda
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