jax.checkpoint_policies.dots_with_no_batch_dims_saveable

checkpoint_policies.dots_with_no_batch_dims_saveable(*args, **params)

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

返回类型:

bool