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

update Evox releated for EvoX-v0.9.1

Browse Source
This commit is contained in:
wls2002
2025-03-26 10:05:49 +08:00
parent 55f1c626d3
commit f017e8357a
3 changed files with 20 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/with_evox/walker2d_evox.py
View File

@@ -48,7 +48,7 @@ def nan2inf(x):
workflow = workflows.StdWorkflow(
algorithm=evox_algorithm,
problem=problem,
candidate_transforms=[jax.jit(jax.vmap(evox_algorithm.transform))],
solution_transforms=[jax.jit(jax.vmap(evox_algorithm.transform))],
fitness_transforms=[nan2inf],
monitors=[monitor],
opt_direction="max",
@@ -62,5 +62,5 @@ state = workflow.enable_multi_devices(state)
# run the workflow for 100 steps
for i in range(100):
train_info, state = workflow.step(state)
state = workflow.step(state)
monitor.show()