package owl

  1. Overview
  2. Docs
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module D.CheckpointSource

Sourcetype state = Owl_optimise_generic.Make(Owl_algodiff_generic.Make(Owl_algodiff_primal_ops.D)).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;
}
Sourcetype typ = Owl_optimise_generic.Make(Owl_algodiff_generic.Make(Owl_algodiff_primal_ops.D)).Checkpoint.typ =
  1. | Batch of int
  2. | Epoch of float
  3. | Custom of state -> unit
  4. | None
Sourceval init_state : int -> float -> state
Sourceval default_checkpoint_fun : (string -> 'a) -> 'a
Sourceval print_state_info : state -> unit
Sourceval print_summary : state -> unit
Sourceval run : typ -> (string -> unit) -> int -> Algodiff.t -> state -> unit
Sourceval to_string : typ -> string
OCaml

Innovation. Community. Security.