Module Optimise.Checkpoint

type state = Owl_neural_neuron.Make(Owl_optimise_generic.Make(Owl_algodiff_generic.Make(A))).Optimise.Checkpoint.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;
}
val init_state : int -> float -> state
val default_checkpoint_fun : (string -> 'a) -> 'a
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