sengi commited on
Commit
ded9faf
·
verified ·
1 Parent(s): 70a313e

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -46,7 +46,7 @@
46
  ]
47
  }
48
  },
49
- "device": "cpu",
50
  "use_amp": false,
51
  "chunk_size": 50,
52
  "n_action_steps": 50,
 
46
  ]
47
  }
48
  },
49
+ "device": "cuda",
50
  "use_amp": false,
51
  "chunk_size": 50,
52
  "n_action_steps": 50,