Inheritance diagram for dismec::init::PreTrainedInitializationStrategy:

Public Member Functions
	PreTrainedInitializationStrategy (std::shared_ptr< const model::Model > pre_trained)

std::unique_ptr< WeightsInitializer >	make_initializer (const std::shared_ptr< const GenericFeatureMatrix > &features) const override
	Creats a new, thread local `WeightsInitializer`. More...

Public Member Functions inherited from dismec::init::WeightInitializationStrategy
virtual	~WeightInitializationStrategy ()=default

Private Attributes
std::shared_ptr< const model::Model >	m_PreTrained

Detailed Description

Definition at line 31 of file pretrained.cpp.

Constructor & Destructor Documentation

◆ PreTrainedInitializationStrategy()

PreTrainedInitializationStrategy::PreTrainedInitializationStrategy ( std::shared_ptr< const model::Model > pre_trained )

Definition at line 40 of file pretrained.cpp.

Member Function Documentation

◆ make_initializer()

std::unique_ptr< WeightsInitializer > PreTrainedInitializationStrategy::make_initializer ( const std::shared_ptr< const GenericFeatureMatrix > & features ) const

overridevirtual

Creats a new, thread local WeightsInitializer.

This function will be called from the thread in which the returned WeightsInitializer will be used. It gets passed in a numa-local copy of the feature matrix.

Parameters

features Read-only reference to the numa-local feature matrix.