package owl

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

Module Owl_neural_parallel.MakeSource

Parameters

module M : ModelSig
module E : EngineSig

Signature

Sourcetype task = {
  1. mutable id : int;
  2. mutable state : Owl_optimise.S.Checkpoint.state option;
  3. mutable params : Owl_optimise.S.Params.typ;
  4. mutable model : M.network;
  5. mutable data_x : Owl_algodiff.S.t;
  6. mutable data_y : Owl_algodiff.S.t;
}
Sourceval delta_model : M.network -> M.network -> unit
Sourceval local_model : task -> 'a
Sourceval schedule : task -> 'a list -> ('a * (int * 'b) list) list
Sourceval pull : task -> ('a * M.network) list -> ('a * 'b) list
Sourceval push : task -> 'a -> ('b * M.network) list -> ('b * M.network) list
Sourceval stop : 'a -> 'b -> bool
Sourceval train_generic : ?params:Owl_optimise.S.Params.typ -> M.network -> Owl_algodiff.S.t -> Owl_algodiff.S.t -> string -> string -> unit
Sourceval train : ?params:Owl_optimise.S.Params.typ -> M.network -> Owl_algodiff.S.A.arr -> Owl_algodiff.S.A.arr -> string -> string -> unit
OCaml

Innovation. Community. Security.

GitHub Discord Twitter Peertube RSS
About
Changelog Releases Industrial Users Academic Users Why OCaml
Ecosystem
Packages Community Events OCaml Planet Jobs
Resources
Install OCaml Get Started Platform Tools Language Manual Standard Library API Books Exercises Papers OCaml Playground Logo
Policies
Carbon Footprint Governance Privacy Code of Conduct