Module Owl_optimise_generic.Make

Parameters

Signature

module Algodiff = Algodiff
module Utils : sig ... end
module Learning_Rate : sig ... end
module Batch : sig ... end
module Loss : sig ... end
module Gradient : sig ... end
module Momentum : sig ... end
module Regularisation : sig ... end
module Clipping : sig ... end
module Stopping : sig ... end
module Checkpoint : sig ... end
module Params : sig ... end
val minimise_network : ?state:Checkpoint.state -> Params.typ -> (Algodiff.t -> Algodiff.t * Algodiff.t array array) -> (Algodiff.t -> Algodiff.t array array * Algodiff.t array array) -> (Algodiff.t array array -> unit) -> (string -> unit) -> Algodiff.t -> Algodiff.t -> Checkpoint.state
val minimise_compiled_network : ?state:Checkpoint.state -> Params.typ -> (Algodiff.t -> Algodiff.t -> Algodiff.t) -> (unit -> unit) -> (string -> unit) -> Algodiff.t -> Algodiff.t -> Checkpoint.state