package torch

  1. Overview
  2. Docs
type t
module Clip_grad : sig ... end
val adam : ?beta1:float -> ?beta2:float -> ?weight_decay:float -> Var_store.t -> learning_rate:float -> t
val rmsprop : ?alpha:float -> ?eps:float -> ?weight_decay:float -> ?momentum:float -> ?centered:bool -> Var_store.t -> learning_rate:float -> t
val sgd : ?momentum:float -> ?dampening:float -> ?weight_decay:float -> ?nesterov:bool -> Var_store.t -> learning_rate:float -> t
val step : ?clip_grad:Clip_grad.t -> t -> unit
val zero_grad : t -> unit
val backward_step : ?clip_grad:Clip_grad.t -> t -> loss:Tensor.t -> unit
val set_learning_rate : t -> learning_rate:float -> unit
module Linear_interpolation : sig ... end
OCaml

Innovation. Community. Security.