Interface GradCheckpointTraining

enable

• enable(
      targets: readonly { lw: LoraLeaves; modulePath: string }[],
      splitMlp: boolean,
  ): GradCheckpointRun

  Enable for the adapter's targets; splitMlp also checkpoints each layer's attention and MLP sub-blocks separately.

  Parameters

  • targets: readonly { lw: LoraLeaves; modulePath: string }[]
  • splitMlp: boolean

  Returns GradCheckpointRun