Skip to content
Snippets Groups Projects

remove unused observation and prediction files

Merged Christian Baumberger requested to merge 2-cleanup-baselines into master
1 unresolved thread
Files
12
+ 2
2
@@ -28,8 +28,8 @@ test_dones = []
sequential_agent_test = False
# Load your agent
agent = Agent(state_size, action_size, 0)
agent.qnetwork_local.load_state_dict(torch.load('../torch_training/Nets/avoid_checkpoint60000.pth'))
agent = Agent(state_size, action_size)
agent.qnetwork_local.load_state_dict(torch.load('../torch_training/Nets/avoid_checkpoint500.pth'))
# Load the necessary Observation Builder and Predictor
predictor = ShortestPathPredictorForRailEnv()
Loading