add brax env

2023-10-17 20:20:03 +08:00
parent f217d87ac6
commit 7f042e07c2
9 changed files with 201 additions and 6 deletions
--- a/pipeline.py
+++ b/pipeline.py
@@ -24,6 +24,8 @@ class Pipeline:
        self.algorithm = algorithm
        self.problem = problem_type(config.problem)

+        print(self.problem.input_shape, self.problem.output_shape)
+
        if isinstance(algorithm, NEAT):
            assert config.neat.inputs == self.problem.input_shape[-1], f"problem input shape {self.problem.input_shape}"