package torch

  1. Overview
  2. Docs
type t
val adam : learning_rate:float -> beta1:float -> beta2:float -> weight_decay:float -> t
val rmsprop : learning_rate:float -> alpha:float -> eps:float -> weight_decay:float -> momentum:float -> centered:bool -> t
val sgd : learning_rate:float -> momentum:float -> dampening:float -> weight_decay:float -> nesterov:bool -> t
val set_learning_rate : t -> float -> unit
val set_momentum : t -> float -> unit
val add_parameters : t -> Tensor.t list -> unit
val zero_grad : t -> unit
val step : t -> unit
OCaml

Innovation. Community. Security.