Skip to content
Snippets Groups Projects
Commit 526c6dfe authored by xarthur's avatar xarthur
Browse files

fix cpu/gpu display bug.

parent b906e17d
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment