Expose original Atari game reward

2026-09-09 11:13:47 +08:00 · 2017-12-23 20:52:17 -07:00
parent 69b7ea53cc
commit 964734755f
9 changed files with 11 additions and 8 deletions
@@ -37,6 +37,7 @@ class Config:
        self.noise_decay_interval = 0
        self.target_network_mix = 0.001
        self.action_shift_fn = lambda a: a
+        self.reward_shift_fn = lambda r: r
        self.reward_weight = 1
        self.hybrid_reward = False
        self.target_type = self.q_target