tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix bugs

Browse Source
This commit is contained in:
root
2024-07-12 02:25:57 +08:00
parent 3194678a15
commit 5fc63fdaf1
28 changed files with 351 additions and 142 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
examples/brax/halfcheetah.py
View File

@@ -3,7 +3,7 @@ from tensorneat.algorithm.neat import NEAT
from tensorneat.genome import DefaultGenome, BiasNode
from tensorneat.problem.rl import BraxEnv
from tensorneat.common import Act, Agg
from tensorneat.common import ACT, AGG
import jax
@@ -26,10 +26,10 @@ if __name__ == "__main__":
num_outputs=6,
init_hidden_layers=(),
node_gene=BiasNode(
activation_options=Act.tanh,
aggregation_options=Agg.sum,
activation_options=ACT.tanh,
aggregation_options=AGG.sum,
),
output_transform=Act.standard_tanh,
output_transform=ACT.standard_tanh,
),
),
problem=BraxEnv(

8
examples/brax/walker2d.py
View File

@@ -3,7 +3,7 @@ from tensorneat.algorithm.neat import NEAT
from tensorneat.genome import DefaultGenome, BiasNode
from tensorneat.problem.rl import BraxEnv
from tensorneat.common import Act, Agg
from tensorneat.common import ACT, AGG
import jax, jax.numpy as jnp
@@ -26,10 +26,10 @@ if __name__ == "__main__":
num_outputs=6,
init_hidden_layers=(),
node_gene=BiasNode(
activation_options=Act.tanh,
aggregation_options=Agg.sum,
activation_options=ACT.tanh,
aggregation_options=AGG.sum,
),
output_transform=Act.standard_tanh,
output_transform=ACT.standard_tanh,
),
),
problem=BraxEnv(