Class Sgd

• Defined in File optimizers.h

Inheritance Relationships

Base Type

• public marian::OptimizerBase (Class OptimizerBase)

Class Documentation

class Sgd : public marian::OptimizerBase

Stochastic gradient descent optimizer.

Public Functions

Sgd(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)

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