dl::ModelBase Class Reference

Inheritance diagram for dl::ModelBase:

Public Member Functions
size_t	numParameters () const noexcept
size_t	numTrainableParams () const noexcept
std::map< std::string, dl::TensorRef > &	parameters () noexcept
const std::map< std::string, dl::TensorRef > &	parameters () const noexcept

Protected Member Functions
void	registerParameter (std::string name, TensorPtr &tensor)
void	registerParameters (std::string prefix, std::ranges::range auto &tensors)

Detailed Description

Definition at line 13 of file model.hpp.

Member Function Documentation

parameters() [1/2]

const std::map< std::string, dl::TensorRef > & dl::ModelBase::parameters ( ) const

inlinenoexcept

Definition at line 29 of file model.hpp.

{ return _parameters; }

parameters() [2/2]

std::map< std::string, dl::TensorRef > & dl::ModelBase::parameters ( )

inlinenoexcept

Definition at line 28 of file model.hpp.

{ return _parameters; }

registerParameters()

void dl::ModelBase::registerParameters ( std::string  prefix, std::ranges::range auto &  tensors  )

inlineprotected

Definition at line 19 of file model.hpp.

                                                                                {
            for (auto&& [key, value] : tensors)
                _parameters.insert({std::format("{}.{}", prefix, key), value});
        }

---

The documentation for this class was generated from the following file:

• dl/model/model.hpp