wassname/Run-Skeleton-Run
1
0
Fork 0
mirror of https://github.com/wassname/Run-Skeleton-Run.git synced 2026-08-04 12:45:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix norm

Browse Source
This commit is contained in:
wassname
2018-01-21 12:45:56 +08:00
parent 11cb4d9cce
commit f4c45b03db
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ddpg/model.py
+2 -2
View File
@@ -216,7 +216,7 @@ def train_multi_thread(actor, critic, dynamics, target_actor, target_critic, tar
env = create_env(args)
random_process = create_random_process(args)
state_normalizer = StaticNormalizer(env.state_dim)
state_normalizer = StaticNormalizer(int(np.prod(env.observation_space.shape)))
reward_normalizer = StaticNormalizer(1)
actor_learning_rate_decay_fn = create_decay_fn(
@@ -488,7 +488,7 @@ def play_single_thread(
num_cycles=args.max_episodes // epsilon_cycle_len)
env = create_env(args)
state_normalizer = StaticNormalizer(env.state_dim)
state_normalizer = StaticNormalizer(int(np.prod(env.observation_space.shape)))
reward_normalizer = StaticNormalizer(1)
episode = 1