From 4f39f877862c7334e1c2dc13cc1751ea7c2626ef Mon Sep 17 00:00:00 2001
From: kowalcorp <50779494+kowalcorp@users.noreply.github.com>
Date: Mon, 9 Jan 2023 17:32:56 +0100
Subject: [PATCH] small corrections in row 54

---
 .../PolicyGradient/actor_critic/actor_critic_replay_torch.py    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ReinforcementLearning/PolicyGradient/actor_critic/actor_critic_replay_torch.py b/ReinforcementLearning/PolicyGradient/actor_critic/actor_critic_replay_torch.py
index f6fac17..56ba73d 100644
--- a/ReinforcementLearning/PolicyGradient/actor_critic/actor_critic_replay_torch.py
+++ b/ReinforcementLearning/PolicyGradient/actor_critic/actor_critic_replay_torch.py
@@ -51,7 +51,7 @@ def __init__(self, lr, input_dims, fc1_dims, fc2_dims,
         self.v = nn.Linear(self.fc2_dims, 1)
         self.optimizer = optim.Adam(self.parameters(), lr=lr)
 
-        self.device = T.device('cuda:0' if T.cuda.is_available() else 'cuda:1')
+        self.device = T.device('cuda:0' if T.cuda.is_available() else 'cpu')
         self.to(self.device)
 
     def forward(self, state):