Module Make.Checkpoint

type state = {
  1. mutable current_batch : int;
  2. mutable batches_per_epoch : int;
  3. mutable epochs : float;
  4. mutable batches : int;
  5. mutable loss : Algodiff.t array;
  6. mutable start_at : float;
  7. mutable stop : bool;
  8. mutable gs : Algodiff.t array array;
  9. mutable ps : Algodiff.t array array;
  10. mutable us : Algodiff.t array array;
  11. mutable ch : Algodiff.t array array array;
}
type typ =
  1. | Batch of int
  2. | Epoch of float
  3. | Custom of state -> unit
  4. | None
val init_state : int -> float -> state
val default_checkpoint_fun : (string -> 'a) -> 'b
val print_state_info : state -> unit
val print_summary : state -> unit
val run : typ -> (string -> unit) -> int -> Algodiff.t -> state -> unit
val to_string : typ -> string