package torch

  1. Overview
  2. Docs

Module Wrapper.OptimizerSource

Sourcetype t
Sourceval adam : learning_rate:float -> beta1:float -> beta2:float -> weight_decay:float -> t
Sourceval rmsprop : learning_rate:float -> alpha:float -> eps:float -> weight_decay:float -> momentum:float -> centered:bool -> t
Sourceval sgd : learning_rate:float -> momentum:float -> dampening:float -> weight_decay:float -> nesterov:bool -> t
Sourceval set_learning_rate : t -> float -> unit
Sourceval set_momentum : t -> float -> unit
Sourceval add_parameters : t -> Tensor.t list -> unit
Sourceval zero_grad : t -> unit
Sourceval step : t -> unit
OCaml

Innovation. Community. Security.