From 526c6dfef3b4058bcccaabe8d03761ed7ca7940f Mon Sep 17 00:00:00 2001
From: xarthur <xliotx@gmail.com>
Date: Wed, 14 Aug 2019 17:05:22 +0200
Subject: [PATCH] fix cpu/gpu display bug.
---
torch_training/dueling_double_dqn.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/torch_training/dueling_double_dqn.py b/torch_training/dueling_double_dqn.py
index dd67b4f..4c2f0fa 100644
--- a/torch_training/dueling_double_dqn.py
+++ b/torch_training/dueling_double_dqn.py
@@ -20,7 +20,6 @@ double_dqn = True # If using double dqn algorithm
input_channels = 5 # Number of Input channels
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-device = torch.device("cpu")
print(device)
--
GitLab