third commit

RL-MVA-2023-24 · Mar 2, 2024 · a2fe4fb · a2fe4fb
1 parent 37983ae
commit a2fe4fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/train.py b/src/train.py
@@ -50,7 +50,7 @@ class ProjectAgent:
     def __init__(self):
         self.env = TimeLimit(env=HIVPatient(domain_randomization=False), max_episode_steps=200)
         self.path = os.path.join(os.getcwd(),'best_agent.pth')
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.device = torch.device('cpu')
         self.Q = DQN(6, 4).to(self.device)
         self.Q_target = deepcopy(self.Q)
         self.criterion = torch.nn.MSELoss()