From dc6df64492012362dfa955ab103226560411bb55 Mon Sep 17 00:00:00 2001
From: xArthur <xliotx@gmail.com>
Date: Wed, 11 Sep 2019 23:40:41 +0200
Subject: [PATCH] fix path error when executing from cmd.
---
torch_training/dueling_double_dqn.py | 2 +-
torch_training/multi_agent_training.py | 5 +++++
torch_training/training_navigation.py | 5 +++++
3 files changed, 11 insertions(+), 1 deletion(-)
diff --git a/torch_training/dueling_double_dqn.py b/torch_training/dueling_double_dqn.py
index dd67b4f..cf2f7d5 100644
--- a/torch_training/dueling_double_dqn.py
+++ b/torch_training/dueling_double_dqn.py
@@ -20,7 +20,7 @@ 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")
+#device = torch.device("cpu")
print(device)
diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index ad42e0a..222430d 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -4,6 +4,11 @@ import random
import sys
from collections import deque
+# make sure the root path is in system path
+from pathlib import Path
+base_dir = Path(__file__).resolve().parent.parent
+sys.path.append(str(base_dir))
+
import matplotlib.pyplot as plt
import numpy as np
import torch
diff --git a/torch_training/training_navigation.py b/torch_training/training_navigation.py
index c97f1f5..f69929f 100644
--- a/torch_training/training_navigation.py
+++ b/torch_training/training_navigation.py
@@ -3,6 +3,11 @@ import random
import sys
from collections import deque
+# make sure the root path is in system path
+from pathlib import Path
+base_dir = Path(__file__).resolve().parent.parent
+sys.path.append(str(base_dir))
+
import matplotlib.pyplot as plt
import numpy as np
import torch
--
GitLab