Skip to content

R
REST
Commit 96539d53 by lvzhengyang
Options

check VectorEnv

parent f26f7266
Showing with 1 additions and 0 deletions
train.py
...@@ -19,6 +19,7 @@ env_fns = [lambda: create_RSMTEnv(num_nodes=num_nodes, ...@@ -19,6 +19,7 @@ env_fns = [lambda: create_RSMTEnv(num_nodes=num_nodes,
for i in range (batch_size) for i in range (batch_size)
] ]
envs = gym.vector.AsyncVectorEnv(env_fns) envs = gym.vector.AsyncVectorEnv(env_fns)
pdb.set_trace()
actor_net = Actor() actor_net = Actor()
critic_net = Critic() critic_net = Critic()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment