dismec::init::PreTrainedInitializer Class Reference
Inheritance diagram for dismec::init::PreTrainedInitializer:
Public Member Functions | |
| PreTrainedInitializer (std::shared_ptr< const model::Model > pre_trained) | |
| void | get_initial_weight (label_id_t label_id, Eigen::Ref< DenseRealVector > target, objective::Objective &objective) override |
| Generate an initial vector for the given label. The result should be placed in target. More... | |
 Public Member Functions inherited from dismec::init::WeightsInitializer | |
| virtual | ~WeightsInitializer ()=default |
| Public Member Functions inherited from dismec::stats::Tracked | |
| Tracked () | |
Default constructor, creates the internal stats::StatisticsCollection. More... | |
| void | register_stat (const std::string &name, std::unique_ptr< Statistics > stat) |
Registers a tracker for the statistics name. More... | |
| std::shared_ptr< StatisticsCollection > | get_stats () const |
Gets an ownership-sharing reference to the StatisticsCollection. More... | |
Private Attributes | |
| std::shared_ptr< const model::Model > | m_PreTrainedWeights |
Additional Inherited Members | |
| Protected Member Functions inherited from dismec::stats::Tracked | |
| ~Tracked () | |
| Non-virtual destructor. Declared protected, so we don't accidentally try to do a polymorphic delete. More... | |
| template<class T > | |
| void | record (stat_id_t stat, T &&value) |
Record statistics. This function just forwards all its arguments to the internal StatisticsCollection. More... | |
| void | declare_stat (stat_id_t index, StatisticMetaData meta) |
Declares a new statistics. This function just forwards all its arguments to the internal StatisticsCollection. More... | |
| void | declare_tag (tag_id_t index, std::string name) |
Declares a new tag. This function just forwards all its arguments to the internal StatisticsCollection. More... | |
| template<class... Args> | |
| void | set_tag (tag_id_t tag, long value) |
Set value of tag. This function just forwards all its arguments to the internal StatisticsCollection. More... | |
| template<class... Args> | |
| auto | make_timer (stat_id_t id, Args... args) |
Creates a new ScopeTimer using stats::record_scope_time. More... | |
Detailed Description
Definition at line 13 of file pretrained.cpp.
Constructor & Destructor Documentation
◆ PreTrainedInitializer()
|
inlineexplicit |
Definition at line 15 of file pretrained.cpp.
References m_PreTrainedWeights.
Member Function Documentation
◆ get_initial_weight()
|
inlineoverridevirtual |
Generate an initial vector for the given label. The result should be placed in target.
Implements dismec::init::WeightsInitializer.
Definition at line 22 of file pretrained.cpp.
References m_PreTrainedWeights.
Member Data Documentation
◆ m_PreTrainedWeights
|
private |
Definition at line 28 of file pretrained.cpp.
Referenced by get_initial_weight(), and PreTrainedInitializer().
The documentation for this class was generated from the following file:
- src/training/init/pretrained.cpp
