gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

add missing hidden layer in rainbow_q_head

Browse Source
This commit is contained in:
Gal Leibovich
2018-08-30 19:34:27 +03:00
parent ea294de7fd
commit ebe574e463
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rl_coach/presets/Atari_Rainbow.py
View File

@@ -15,7 +15,7 @@ schedule_params = ScheduleParameters()
schedule_params.improve_steps = EnvironmentSteps(50000000)
schedule_params.steps_between_evaluation_periods = EnvironmentSteps(250000)
schedule_params.evaluation_steps = EnvironmentSteps(135000)
schedule_params.heatup_steps = EnvironmentSteps(500)
schedule_params.heatup_steps = EnvironmentSteps(50000)
#########
# Agent #