wassname/Run-Skeleton-Run
1
0
Fork 0
mirror of https://github.com/wassname/Run-Skeleton-Run.git synced 2026-08-07 11:18:09 +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/train.py
+3 -1
View File
@@ -5,10 +5,12 @@ import copy
import torch
import torch.multiprocessing as mp
from multiprocessing import Value
import numpy as np
from common.misc_util import boolean_flag, str2params, create_if_need
from common.env_wrappers import create_env
from common.torch_util import activations, hard_update
from common.normalizer import StaticNormalizer
from ddpg.model import create_model, create_act_update_fns, train_multi_thread, \
train_single_thread, play_single_thread
@@ -148,7 +150,7 @@ def train(args, model_fn, act_update_fns, multi_thread, train_single, play_singl
json.dump(vars(args), fout, indent=4, ensure_ascii=False, sort_keys=True)
env = create_env(args)
shared_state_normalizer = StaticNormalizer(env.state_dim)
shared_state_normalizer = StaticNormalizer(int(np.prod(env.observation_space.shape)))
shared_reward_normalizer = StaticNormalizer(1)
if args.flip_state_action and hasattr(env, "state_transform"):