Add examples for saving evolving state and then restore evolving

This commit is contained in:
wls2002
2025-02-22 18:51:02 +08:00
parent 71771debfc
commit 6d5f573e28
4 changed files with 111 additions and 0 deletions

View File

@@ -83,6 +83,13 @@ class Pipeline(StatefulBaseClass):
return state
def step(self, state):
"""
returns:
state, previous_pop, fitnesses
state: updated state
previous_pop: previous population
fitnesses: fitnesses of previous population
"""
randkey_, randkey = jax.random.split(state.randkey)