fix bugs
This commit is contained in:
@@ -4,7 +4,7 @@ from tensorneat.pipeline import Pipeline
|
||||
from tensorneat.algorithm.neat import NEAT
|
||||
from tensorneat.algorithm.hyperneat import HyperNEAT, FullSubstrate
|
||||
from tensorneat.genome import DefaultGenome
|
||||
from tensorneat.common import Act
|
||||
from tensorneat.common import ACT
|
||||
|
||||
from tensorneat.problem import GymNaxEnv
|
||||
|
||||
@@ -27,10 +27,10 @@ if __name__ == "__main__":
|
||||
num_inputs=4, # size of query coors
|
||||
num_outputs=1,
|
||||
init_hidden_layers=(),
|
||||
output_transform=Act.standard_tanh,
|
||||
output_transform=ACT.standard_tanh,
|
||||
),
|
||||
),
|
||||
activation=Act.tanh,
|
||||
activation=ACT.tanh,
|
||||
activate_time=10,
|
||||
output_transform=jnp.argmax,
|
||||
),
|
||||
|
||||
Reference in New Issue
Block a user