OpenSportsLab
/

OSL-loc-snbas-2023-e2e

Model card Files Files and versions

jeetv commited on Aug 3

Commit

42dd642

·

verified ·

1 Parent(s): af0b227

Update config.yaml

Files changed (1) hide show

config.yaml +1 -1

config.yaml CHANGED Viewed

@@ -117,6 +117,6 @@ SYSTEM:
   work_dir: ${SYSTEM.save_dir}
   seed: 42
   GPU: 4         # number of gpus to use
-  device: cuda   # auto | cuda | cpu
   gpu_id: 0      # device id for single gpu training

   work_dir: ${SYSTEM.save_dir}
   seed: 42
   GPU: 4         # number of gpus to use
+  device: auto   # auto | cuda | cpu
   gpu_id: 0      # device id for single gpu training