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/half_cheetah.py
+++ b/examples/brax/half_cheetah.py
@@ -15,7 +15,8 @@ def example_conf():
        basic=BasicConfig(
            seed=42,
            fitness_target=10000,
-            pop_size=10000
+            generation_limit=10,
+            pop_size=100
        ),
        neat=NeatConfig(
            inputs=17,
@@ -33,9 +34,9 @@ def example_conf():

 if __name__ == '__main__':
    conf = example_conf()
-
    algorithm = NEAT(conf, NormalGene)
    pipeline = Pipeline(conf, algorithm, BraxEnv)
    state = pipeline.setup()
    pipeline.pre_compile(state)
    state, best = pipeline.auto_run(state)
+    pipeline.show(state, best, save_path="half_cheetah.gif", )