package torch

  1. Overview
  2. Docs
package torch
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Wrapper.OptimizerSource

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