Class EncoderPooler

• Defined in File encoder_pooler.h

Inheritance Relationships

Base Type

• public marian::EncoderPoolerBase (Class EncoderPoolerBase)

Class Documentation

class EncoderPooler : public marian::EncoderPoolerBase

Public Types

typedef data::Corpus dataset_type

Public Functions

EncoderPooler(Ptr<Options> options)

virtual Ptr<Options> getOptions()

std::vector<Ptr<EncoderBase>> &getEncoders()

std::vector<Ptr<PoolerBase>> &getPoolers()

void push_back(Ptr<EncoderBase> encoder)

void push_back(Ptr<PoolerBase> pooler)

void load(Ptr<ExpressionGraph> graph, const std::string &name, bool markedReloaded)

void mmap(Ptr<ExpressionGraph> graph, const void *ptr, bool markedReloaded)

void save(Ptr<ExpressionGraph> graph, const std::string &name, bool)

void clear(Ptr<ExpressionGraph> graph)

template<typename T>
T opt(const std::string &key)

template<typename T>
T opt(const std::string &key, const T &def)

template<typename T>
void set(std::string key, T value)

virtual std::vector<Expr> apply(Ptr<ExpressionGraph> graph, Ptr<data::CorpusBatch> batch, bool clearGraph)

Protected Functions

Config::YamlNode getModelParameters()

std::string getModelParametersAsString()

Protected Attributes

Ptr<Options> options_

std::string prefix_

std::vector<Ptr<EncoderBase>> encoders_

std::vector<Ptr<PoolerBase>> poolers_

bool inference_ = {true}

std::set<std::string> modelFeatures_