bacow/easy-rl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update rainbowdqn

Browse Source
This commit is contained in:
johnjim0816
2022-05-31 01:20:58 +08:00
parent cfc0f6492e
commit c7c94468c9
149 changed files with 1866 additions and 1549 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
codes/SoftActorCritic/task0.py
View File

@@ -63,9 +63,9 @@ class PlotConfig:
def env_agent_config(cfg,seed=1):
env = NormalizedActions(gym.make(cfg.env_name))
env.seed(seed)
action_dim = env.action_space.shape[0]
state_dim = env.observation_space.shape[0]
agent = SAC(state_dim,action_dim,cfg)
n_actions = env.action_space.shape[0]
n_states = env.observation_space.shape[0]
agent = SAC(n_states,n_actions,cfg)
return env,agent
def train(cfg,env,agent):