conatus.expert.revolve_cost

conatus.expert.revolve_cost(steps, checkpoints)[source]

Forward steps recomputed by the reverse sweep of FlowMap.vjp (excluding the stage recomputation inside each step’s adjoint), its ratio to steps, the peak number of stored checkpoints and the number of checkpoint stores. checkpoints=None: full storage.

Parameters:
  • steps (int)

  • checkpoints (int | None)

Return type:

dict