mirror of
https://github.com/gryf/coach.git
synced 2025-12-18 03:30:19 +01:00
QR-DQN bug fix and imporvements (#30)
* bug fix - QR-DQN using error instead of abs-error in the quantile huber loss * improvement - QR-DQN sorting the quantile only once instead of batch_size times * new feature - adding the Breakout QRDQN preset (verified to achieve good results)
This commit is contained in:
18
presets.py
18
presets.py
@@ -422,6 +422,24 @@ class Breakout_C51(Preset):
|
||||
self.evaluate_every_x_episodes = 5000000
|
||||
|
||||
|
||||
|
||||
class Breakout_QRDQN(Preset):
|
||||
def __init__(self):
|
||||
Preset.__init__(self, QuantileRegressionDQN, Atari, ExplorationParameters)
|
||||
self.env.level = 'BreakoutDeterministic-v4'
|
||||
self.agent.num_steps_between_copying_online_weights_to_target = 10000
|
||||
self.learning_rate = 0.00025
|
||||
self.agent.num_transitions_in_experience_replay = 1000000
|
||||
self.exploration.initial_epsilon = 1.0
|
||||
self.exploration.final_epsilon = 0.01
|
||||
self.exploration.epsilon_decay_steps = 1000000
|
||||
self.exploration.evaluation_policy = 'EGreedy'
|
||||
self.exploration.evaluation_epsilon = 0.001
|
||||
self.num_heatup_steps = 50000
|
||||
self.evaluation_episodes = 1
|
||||
self.evaluate_every_x_episodes = 50
|
||||
|
||||
|
||||
class Atari_DQN_TestBench(Preset):
|
||||
def __init__(self):
|
||||
Preset.__init__(self, DQN, Atari, ExplorationParameters)
|
||||
|
||||
Reference in New Issue
Block a user