Flatten (owl-base.Owl_neural

Parameters

module Graph : Owl_neural_graph_sig.Sig

module Graph = Graph

module Optimise = Graph.Neuron.Optimise

module Algodiff = Graph.Neuron.Optimise.Algodiff

module Init = Graph.Neuron.Init

module Activation = Graph.Neuron.Activation

module Params = Graph.Neuron.Optimise.Params

module Batch = Graph.Neuron.Optimise.Batch

module Learning_Rate = Graph.Neuron.Optimise.Learning_Rate

module Loss = Graph.Neuron.Optimise.Loss

module Gradient = Graph.Neuron.Optimise.Gradient

module Momentum = Graph.Neuron.Optimise.Momentum

module Regularisation = Graph.Neuron.Optimise.Regularisation

module Clipping = Graph.Neuron.Optimise.Clipping

module Stopping = Graph.Neuron.Optimise.Stopping

module Checkpoint = Graph.Neuron.Optimise.Checkpoint