Improve normalizer for scalar

This commit is contained in:
Shangtong Zhang committed 2017-10-07 21:23:02 -06:00
1 parent fae9a85f31
commit 51f125c01e
3 files changed
+13 -4

No files matched your search

+1 -1
View File
@@ -48,7 +48,7 @@ class ContinuousAdvantageActorCritic:
steps += 1
total_reward += reward
reward = np.asscalar(self.reward_normalizer(np.array([reward])))
reward = self.reward_normalizer(reward)
if deterministic:
if terminal:
+1 -1
View File
@@ -79,7 +79,7 @@ class ProximalPolicyOptimization:
batched_steps += 1
episode_length += 1
reward = np.asscalar(self.reward_normalizer(np.array([reward])))
reward = self.reward_normalizer(reward)
rewards.append(reward)
if done: