Skip to content

Commit 393be07

Browse files
ViktorMdementrock
authored andcommitted
Fixed mistypes in hidden layers sizes in TRPO examples (rll#149)
1 parent 6d7df9d commit 393be07

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/trpo_gym_cartpole.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ def run_task(*_):
1616
policy = CategoricalMLPPolicy(
1717
env_spec=env.spec,
1818
# The neural network policy should have two hidden layers, each with 32 hidden units.
19-
hidden_sizes=(8, 8)
19+
hidden_sizes=(32, 32)
2020
)
2121

2222
baseline = LinearFeatureBaseline(env_spec=env.spec)

examples/trpo_gym_pendulum.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def run_task(*_):
1515
policy = GaussianMLPPolicy(
1616
env_spec=env.spec,
1717
# The neural network policy should have two hidden layers, each with 32 hidden units.
18-
hidden_sizes=(8, 8)
18+
hidden_sizes=(32, 32)
1919
)
2020

2121
baseline = LinearFeatureBaseline(env_spec=env.spec)

0 commit comments

Comments
 (0)