fix path error when executing from cmd.

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)
 
 

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

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