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
1 merge request!242 run baselines in ci
Pipeline #1407 canceled
Hide whitespace changes
Inline Side-by-side
torch_training/multi_agent_training.py
+ 1
− 1
View file @ e57ff31c
......@@ -7,7 +7,7 @@ import random
import torch
from dueling_double_dqn import Agent
import torch_training
import torch_training.Nets
from flatland.envs.generators import complex_rail_generator
from flatland.envs.observations import TreeObsForRailEnv
from flatland.envs.predictions import ShortestPathPredictorForRailEnv
......
torch_training/training_navigation.py
+ 1
− 1
View file @ e57ff31c
......@@ -8,7 +8,7 @@ import numpy as np
import torch
from dueling_double_dqn import Agent
import torch_training
import torch_training.Nets
from flatland.envs.generators import complex_rail_generator
from flatland.envs.observations import TreeObsForRailEnv
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