Class Adagrad

• Defined in File optimizers.h

Inheritance Relationships

Base Type

• public marian::OptimizerBase (Class OptimizerBase)

Class Documentation

class Adagrad : public marian::OptimizerBase

Adagrad optimizer.

http://www.jmlr.org/papers/volume12/duchi11a/duchi11a.pdf

Public Functions

Adagrad(Ptr<Options> options)

void load(std::vector<io::Item> &items, const std::vector<Ptr<OptimizerBase>> &opts, const std::vector<Ptr<Backend>> &backends, const ScatterStateFunc &scatterFn, bool isMainProcess)

void save(std::vector<io::Item> &items, const std::vector<Ptr<OptimizerBase>> &opts, const GatherStateFunc &gatherFn, bool isMainProcess)

void setParams(const std::vector<float> &params)

std::vector<Tensor> getShards()