From 899b0f728c2ac104ed1e2df5884917a4c3f54cca Mon Sep 17 00:00:00 2001
From: xzhaoma <ma@arch.ethz.ch>
Date: Thu, 12 Sep 2019 21:57:13 +0000
Subject: [PATCH] fix path error when executing from cmd.
---
.gitignore | 2 ++
torch_training/dueling_double_dqn.py | 2 +-
torch_training/multi_agent_training.py | 5 +++++
torch_training/training_navigation.py | 5 +++++
4 files changed, 13 insertions(+), 1 deletion(-)
diff --git a/.gitignore b/.gitignore
index 9074321..72b8550 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
*pycache*
*ppo_policy*
+
+torch_training/Nets/
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