Skip to content
Snippets Groups Projects
Commit e57ff31c authored by u214892's avatar u214892
Browse files

#42 run baselines in ci

parent eb71fa51
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
torch_training/multi_agent_training.py
+ 1
− 1
View file @ e57ff31c
...@@ -7,7 +7,7 @@ import random ...@@ -7,7 +7,7 @@ import random
import torch import torch
from dueling_double_dqn import Agent from dueling_double_dqn import Agent
import torch_training import torch_training.Nets
from flatland.envs.generators import complex_rail_generator from flatland.envs.generators import complex_rail_generator
from flatland.envs.observations import TreeObsForRailEnv from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv from flatland.envs.predictions import ShortestPathPredictorForRailEnv
......
torch_training/training_navigation.py
+ 1
− 1
View file @ e57ff31c
...@@ -8,7 +8,7 @@ import numpy as np ...@@ -8,7 +8,7 @@ import numpy as np
import torch import torch
from dueling_double_dqn import Agent from dueling_double_dqn import Agent
import torch_training import torch_training.Nets
from flatland.envs.generators import complex_rail_generator from flatland.envs.generators import complex_rail_generator
from flatland.envs.observations import TreeObsForRailEnv from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv from flatland.envs.predictions import ShortestPathPredictorForRailEnv
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment