maxzoech

import orbax.checkpoint as ocp
from orbax.checkpoint.checkpoint_managers import preservation_policy
from orbax.checkpoint.checkpoint_managers import save_decision_policy

options = ocp.CheckpointManagerOptions(
    save_decision_policy=save_decision_policy.FixedIntervalPolicy(100),
    preservation_policy=preservation_policy.AnyPreservationPolicy(
        [
            preservation_policy.LatestN(1),
            preservation_policy.EveryNSteps(1_000),