package owl

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

Parameters

Signature

type task = Owl_neural_parallel.Make(Graph)(E).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 : Graph.network;
  5. mutable data_x : Owl_optimise.S.t;
  6. mutable data_y : Owl_optimise.S.t;
}
val delta_model : Graph.network -> Graph.network -> unit
val local_model : task -> 'a
val schedule : task -> 'a list -> ('a * (int * 'b) list) list
val pull : task -> ('a * Graph.network) list -> ('a * 'b) list
val push : task -> 'a -> ('b * Graph.network) list -> ('b * Graph.network) list
val stop : 'a -> 'b -> bool
val train_generic : ?params:Owl_optimise.S.Params.typ -> Graph.network -> Owl_optimise.S.t -> Owl_optimise.S.t -> string -> string -> unit
val train : ?params:Owl_optimise.S.Params.typ -> Graph.network -> Owl_optimise.S.arr -> Owl_optimise.S.arr -> string -> string -> unit