jax.checkpoint_policies.save_only_these_names

checkpoint_policies.save_only_these_names()

仅保存命名的值，并且仅在给定的名称中保存。