Owl_neural_parallel.Make
type task = {
mutable id : int;
mutable state : Owl_optimise.S.Checkpoint.state option;
mutable params : Owl_optimise.S.Params.typ;
mutable model : M.network;
mutable data_x : Owl_algodiff.S.t;
mutable data_y : Owl_algodiff.S.t;
}
val make_task :
int ->
Owl_optimise.S.Params.typ ->
M.network ->
Owl_algodiff.S.t ->
Owl_algodiff.S.t ->
task
val local_model : task -> 'a
val schedule : task -> 'a list -> ('b * (int * 'c) list) list
val train_generic :
?params:Owl_optimise.S.Params.typ ->
M.network ->
Owl_algodiff.S.t ->
Owl_algodiff.S.t ->
string ->
string ->
unit
val train :
?params:Owl_optimise.S.Params.typ ->
M.network ->
Owl_algodiff.S.A.arr ->
Owl_algodiff.S.A.arr ->
string ->
string ->
unit