Class Scheduler

• Defined in File scheduler.h

Inheritance Relationships

Base Type

• public marian::TrainingObserver (Class TrainingObserver)

Class Documentation

class Scheduler : public marian::TrainingObserver

Public Functions

Scheduler(Ptr<Options> options, Ptr<TrainingState> state, Ptr<IMPIWrapper> mpi = nullptr)

bool isDynamicMBSizeScaling() const

double getDynamicMBSizeMultiplier() const

std::tuple<size_t, float, float> getGradientNormStats() const

std::tuple<size_t, float, float> getLogGradientNormStats() const

bool keepGoing()

void increaseEpoch()

void started()

void finished()

void addValidator(Ptr<ValidatorBase> validator)

bool validating()

bool saving()

bool syncing()

void validate(const std::vector<Ptr<ExpressionGraph>> &graphs, bool isFinal = false)

size_t stalled()

size_t stalled1st()

size_t stalledMax()

size_t stalledMin()

void update(StaticLoss rationalLoss, Ptr<data::Batch> batch)

void update(StaticLoss rationalLoss, size_t numReadBatches, size_t batchSize, size_t batchLabels, float gradientNorm)

void load(const std::string &name)

void save(const std::string &name)

size_t numberOfBatches()

void registerTrainingObserver(Ptr<TrainingObserver> observer)

void actAfterEpoch(TrainingState &state)

void actAfterBatches(TrainingState &state)

void actAfterStalled(TrainingState &state)