From cd59797be7fd7b6245432e9aa7411f161ceae6a3 Mon Sep 17 00:00:00 2001
From: u214892 <u214892@sbb.ch>
Date: Wed, 10 Jul 2019 14:06:56 +0200
Subject: [PATCH] #42 run baselines in ci
---
torch_training/multi_agent_training.py | 8 +++++++-
1 file changed, 7 insertions(+), 1 deletion(-)
diff --git a/torch_training/multi_agent_training.py b/torch_training/multi_agent_training.py
index 5764f86..4f823be 100644
--- a/torch_training/multi_agent_training.py
+++ b/torch_training/multi_agent_training.py
@@ -6,7 +6,6 @@ from collections import deque
import matplotlib.pyplot as plt
import numpy as np
import torch
-from dueling_double_dqn import Agent
from importlib_resources import path
import torch_training.Nets
@@ -15,8 +14,11 @@ from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv
from flatland.envs.rail_env import RailEnv
from flatland.utils.rendertools import RenderTool
+from torch_training.dueling_double_dqn import Agent
from utils.observation_utils import norm_obs_clip, split_tree
+print("multi_agent_trainging.py (1)")
+
def main(argv):
try:
@@ -219,6 +221,10 @@ def main(argv):
plt.show()
+print("multi_agent_trainging.py (2)")
+
if __name__ == '__main__':
print("main")
main(sys.argv[1:])
+
+print("multi_agent_trainging.py (3)")
--
GitLab