Skip to content

Commit d0536d9

Browse files
Minor changes
1 parent 80d9bf2 commit d0536d9

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/ppo_with_pytorch.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -264,9 +264,7 @@ def train(
264264
env.step(action)
265265

266266
# create the agent
267-
policy_layers = [
268-
env.observation_space.shape[-1], 64, 64, env.action_space.shape[-1] * 2
269-
]
267+
policy_layers = [env.observation_space.shape[-1], 64, 64, env.action_space.shape[-1] * 2]
270268
value_layers = [env.observation_space.shape[-1], 64, 64, 1]
271269
agent = PPOAgent(policy_layers, value_layers, entropy_cost, discounting,
272270
reward_scaling, device)

0 commit comments

Comments
 (0)