jax.checkpoint_policies.nothing_saveable#

checkpoint_policies.nothing_saveable(**__)[源]#

重新计算所有内容,就好像没有使用自定义策略一样。

当使用 jax.remat 而不显式指定策略时,这是生效的策略。

返回类型:

bool