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_
void SetRandomizer(TRand *randomizer) override
int InitWeights(float range, TRand *randomizer) override
void ScaleLayerLearningRate(const char *id, double factor)
void CountAlternators(const Network &other, double *same, double *changed) const override
const STRING & name() const
Plumbing(const STRING &name)
int XScaleFactor() const override
void SetNetworkFlags(uint32_t flags) override
bool Serialize(TFile *fp) const override
~Plumbing() override=default
StaticShape InputShape() const override
STRING spec() const override
bool DeSerialize(TFile *fp) override
void SetEnableTraining(TrainingState state) override
float LayerLearningRate(const char *id) const
void DebugWeights() override
const PointerVector< Network > & stack() const
void ConvertToInt() override
void EnumerateLayers(const STRING *prefix, GenericVector< STRING > *layers) const
int RemapOutputs(int old_no, const std::vector< int > &code_map) override
PointerVector< Network > stack_
void Update(float learning_rate, float momentum, float adam_beta, int num_samples) override
float * LayerLearningRatePtr(const char *id) const
void CacheXScaleFactor(int factor) override
virtual void AddToStack(Network *network)
bool SetupNeedsBackprop(bool needs_backprop) override
Network * GetLayer(const char *id) const