Skip to content
Snippets Groups Projects
Commit 04a942e5 authored by Egli Adrian (IT-SCI-API-PFI)'s avatar Egli Adrian (IT-SCI-API-PFI)
Browse files

bug fixed

parent 0e7aa90c
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 3 additions and 1 deletion
run.py
+ 3
− 1
View file @ 04a942e5
...@@ -45,7 +45,7 @@ from reinforcement_learning.dddqn_policy import DDDQNPolicy ...@@ -45,7 +45,7 @@ from reinforcement_learning.dddqn_policy import DDDQNPolicy
# Print per-step logs # Print per-step logs
VERBOSE = True VERBOSE = True
USE_FAST_TREE_OBS = True USE_FAST_TREEOBS = True
# Checkpoint to use (remember to push it!) # Checkpoint to use (remember to push it!)
checkpoint = "./checkpoints/201124171810-7800.pth" # 18.249244799876152 DEPTH=2 AGENTS=10 checkpoint = "./checkpoints/201124171810-7800.pth" # 18.249244799876152 DEPTH=2 AGENTS=10
...@@ -74,9 +74,11 @@ if USE_FAST_TREEOBS: ...@@ -74,9 +74,11 @@ if USE_FAST_TREEOBS:
def check_is_observation_valid(observation): def check_is_observation_valid(observation):
return True return True
def get_normalized_observation(observation, tree_depth: int, observation_radius=0): def get_normalized_observation(observation, tree_depth: int, observation_radius=0):
return observation return observation
tree_observation = FastTreeObs(max_depth=observation_tree_depth) tree_observation = FastTreeObs(max_depth=observation_tree_depth)
state_size = tree_observation.observation_dim state_size = tree_observation.observation_dim
else: else:
......
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