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/algorithm/hyperneat/hyperneat.py
+++ b/algorithm/hyperneat/hyperneat.py
@@ -105,7 +105,7 @@ class HyperNEATGene:
            values = values.at[input_idx].set(inputs_with_bias)
            nodes_ins = values * weights.T
            values = batch_agg(nodes_ins)  # z = agg(ins)
-            values = values * nodes[:, 2] + nodes[:, 1]  # z = z * response + bias
+            # values = values * nodes[:, 2] + nodes[:, 1]  # z = z * response + bias
            values = batch_act(values)  # z = act(z)
            return values