Class Adagrad

• Defined in File adagrad.h

Inheritance Relationships

Base Type

• public torch::optim::Optimizer (Class Optimizer)

Class Documentation

class Adagrad : public torch::optim::Optimizer

Public Functions

inline explicit Adagrad(const std::vector<OptimizerParamGroup> &param_groups, AdagradOptions defaults = {})

inline explicit Adagrad(std::vector<Tensor> params, AdagradOptions defaults = {})

virtual torch::Tensor step(LossClosure closure = nullptr) override

A loss function closure, which is expected to return the loss value.

virtual void save(serialize::OutputArchive &archive) const override

Serializes the optimizer state into the given archive.

virtual void load(serialize::InputArchive &archive) override

Deserializes the optimizer state from the given archive.