Spaces:

LightChen2333
/

OpenSLU

Runtime error

LightChen2333 commited on Feb 19, 2023

Commit

575471f

•

1 Parent(s): ff323e2

Update config/examples/from_pretrained.yaml

Files changed (1) hide show

config/examples/from_pretrained.yaml CHANGED Viewed

@@ -4,7 +4,7 @@ base:
   name: "OpenSLUv1"
   train: false
   test: true
-  device: cuda
   seed: 42
   epoch_num: 300
   batch_size: 16

   name: "OpenSLUv1"
   train: false
   test: true
+  device: cpu
   seed: 42
   epoch_num: 300
   batch_size: 16