tesseract
5.0.0-alpha-619-ge9db
|
Go to the documentation of this file.
20 #ifndef TESSERACT_LSTM_PLUMBING_H_
21 #define TESSERACT_LSTM_PLUMBING_H_
39 return "Sub-classes of Plumbing must implement spec()!";
62 int RemapOutputs(
int old_no,
const std::vector<int>& code_map)
override;
126 void Update(
float learning_rate,
float momentum,
float adam_beta,
127 int num_samples)
override;
132 double* changed)
const override;
144 #endif // TESSERACT_LSTM_PLUMBING_H_
bool IsPlumbingType() const override
GenericVector< float > learning_rates_
virtual void AddToStack(Network *network)
~Plumbing() override=default
void EnumerateLayers(const STRING *prefix, GenericVector< STRING > *layers) const
Plumbing(const STRING &name)
void CountAlternators(const Network &other, double *same, double *changed) const override
int RemapOutputs(int old_no, const std::vector< int > &code_map) override
STRING spec() const override
void CacheXScaleFactor(int factor) override
PointerVector< Network > stack_
int InitWeights(float range, TRand *randomizer) override
void SetRandomizer(TRand *randomizer) override
void SetEnableTraining(TrainingState state) override
void ConvertToInt() override
float * LayerLearningRatePtr(const char *id) const
Network * GetLayer(const char *id) const
void Update(float learning_rate, float momentum, float adam_beta, int num_samples) override
bool SetupNeedsBackprop(bool needs_backprop) override
void DebugWeights() override
void ScaleLayerLearningRate(const char *id, double factor)
bool DeSerialize(TFile *fp) override
const STRING & name() const
StaticShape InputShape() const override
const PointerVector< Network > & stack() const
int XScaleFactor() const override
float LayerLearningRate(const char *id) const
void SetNetworkFlags(uint32_t flags) override
bool Serialize(TFile *fp) const override