Hugging Face's logo

Spaces:
LightChen2333
/
OpenSLU
Runtime error

App Files Community
LightChen2333 commited on
Commit
575471f
·
1 Parent(s): ff323e2

Update config/examples/from_pretrained.yaml

Browse files
Files changed (1) hide show
  1. config/examples/from_pretrained.yaml +1 -1
config/examples/from_pretrained.yaml CHANGED
@@ -4,7 +4,7 @@ base:
4
  name: "OpenSLUv1"
5
  train: false
6
  test: true
7
- device: cuda
8
  seed: 42
9
  epoch_num: 300
10
  batch_size: 16
 
4
  name: "OpenSLUv1"
5
  train: false
6
  test: true
7
+ device: cpu
8
  seed: 42
9
  epoch_num: 300
10
  batch_size: 16