Skip to content

Commit 6c80564

Browse files
Changed network device to CUDA in config
1 parent d760165 commit 6c80564

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/tqc/config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ network:
4343
activation: relu
4444
default_policy_scale: 1.0
4545
scale_lb: 0.1
46-
device: cpu # TO-DO: change back to CUDA
46+
device: cuda
4747

4848
# logging
4949
logger:

0 commit comments

Comments
 (0)