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

updated gifs in README + fix for multiworker crashes + improved Atari DQN and Dueling DDQN presets

Browse Source
This commit is contained in:
Itai Caspi
2018-08-16 18:23:32 +03:00
parent 8f99409387
commit 1de04d6fee
11 changed files with 19 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rl_coach/presets/Atari_DQN.py
View File

@@ -20,7 +20,8 @@ schedule_params.heatup_steps = EnvironmentSteps(50000)
# Agent #
#########
agent_params = DQNAgentParameters()
agent_params.network_wrappers['main'].learning_rate = 0.00025
# since we are using Adam instead of RMSProp, we adjust the learning rate as well
agent_params.network_wrappers['main'].learning_rate = 0.0001
###############
# Environment #