Update intermediate_source/coding_ddpg.py

Svetlana Karslioglu · web-flow · commit 271e5d339fd4 · 2023-06-12T10:49:56.000-07:00
diff --git a/intermediate_source/coding_ddpg.py b/intermediate_source/coding_ddpg.py
@@ -78,7 +78,7 @@
 device = (
     torch.device("cpu") if torch.cuda.device_count() == 0 else torch.device("cuda:0")
 )
-collector_device = torch.device("cpu")
+collector_device = torch.device("cpu")  # Change the device to ``cuda`` to use CUDA
 
 ###############################################################################
 # TorchRL :class:`~torchrl.objectives.LossModule`

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@`
`78`	`78`	`device = (`
`79`	`79`	`torch.device("cpu") if torch.cuda.device_count() == 0 else torch.device("cuda:0")`
`80`	`80`	`)`
`81`		`-collector_device = torch.device("cpu")`
	`81`	+collector_device = torch.device("cpu") # Change the device to ``cuda`` to use CUDA
`82`	`82`
`83`	`83`	`###############################################################################`
`84`	`84`	# TorchRL :class:`~torchrl.objectives.LossModule`