complete show() in brax env

2023-10-22 21:01:06 +08:00
parent 7f042e07c2
commit 15dadebd7e
11 changed files with 152 additions and 22 deletions
--- a/examples/brax/ant.py
+++ b/examples/brax/ant.py
@@ -12,7 +12,7 @@ def example_conf():
        basic=BasicConfig(
            seed=42,
            fitness_target=10000,
-            pop_size=100
+            pop_size=1000
        ),
        neat=NeatConfig(
            inputs=27,
@@ -23,6 +23,7 @@ def example_conf():
            activation_options=(Act.tanh,),
        ),
        problem=BraxConfig(
+            env_name="ant"
        )
    )