jax.checkpoint_policies.save_from_both_policies

checkpoint_policies.save_from_both_policies(policy_2)

给定策略的逻辑 OR。

一个残差是可保存的，当且仅当它根据任一策略都是可保存的。