diff --git a/torch_training/dueling_double_dqn.py b/torch_training/dueling_double_dqn.py
index dd67b4f0d73ffe1b3f4ad3e947debf18508e78b0..4c2f0faacd5795d73e3e88834ca0762f68e728c0 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)