jax.checkpoint_policies.checkpoint_dots_with_no_batch_dims

checkpoint_policies.checkpoint_dots_with_no_batch_dims(*args, **params)

这是Transformer的一个有用启发式方法。

返回类型:

bool