jax.checkpoint_policies.save_only_these_names

Contents

jax.checkpoint_policies.save_only_these_names#

checkpoint_policies.save_only_these_names()[source]#: Save only named values, and only among the names given.