mlpack 3.4.2
load_output_parameter_visitor.hpp
Go to the documentation of this file.
1
14#ifndef MLPACK_METHODS_ANN_VISITOR_LOAD_OUTPUT_PARAMETER_VISITOR_HPP
15#define MLPACK_METHODS_ANN_VISITOR_LOAD_OUTPUT_PARAMETER_VISITOR_HPP
16
18
19#include <boost/variant.hpp>
20
21namespace mlpack {
22namespace ann {
23
28class LoadOutputParameterVisitor : public boost::static_visitor<void>
29{
30 public:
32 LoadOutputParameterVisitor(std::vector<arma::mat>& parameter);
33
35 template<typename LayerType>
36 void operator()(LayerType* layer) const;
37
38 void operator()(MoreTypes layer) const;
39
40 private:
42 std::vector<arma::mat>& parameter;
43
46 template<typename T>
47 typename std::enable_if<
48 !HasModelCheck<T>::value, void>::type
49 OutputParameter(T* layer) const;
50
53 template<typename T>
54 typename std::enable_if<
55 HasModelCheck<T>::value, void>::type
56 OutputParameter(T* layer) const;
57};
58
59} // namespace ann
60} // namespace mlpack
61
62// Include implementation.
63#include "load_output_parameter_visitor_impl.hpp"
64
65#endif
LoadOutputParameterVisitor restores the output parameter using the given parameter set.
LoadOutputParameterVisitor(std::vector< arma::mat > &parameter)
Restore the output parameter given a parameter set.
void operator()(LayerType *layer) const
Restore the output parameter.
void operator()(MoreTypes layer) const
boost::variant< Linear3D< arma::mat, arma::mat, NoRegularizer > *, Glimpse< arma::mat, arma::mat > *, Highway< arma::mat, arma::mat > *, MultiheadAttention< arma::mat, arma::mat, NoRegularizer > *, Recurrent< arma::mat, arma::mat > *, RecurrentAttention< arma::mat, arma::mat > *, ReinforceNormal< arma::mat, arma::mat > *, Reparametrization< arma::mat, arma::mat > *, Select< arma::mat, arma::mat > *, Sequential< arma::mat, arma::mat, false > *, Sequential< arma::mat, arma::mat, true > *, Subview< arma::mat, arma::mat > *, VRClassReward< arma::mat, arma::mat > *, VirtualBatchNorm< arma::mat, arma::mat > *, RBF< arma::mat, arma::mat, GaussianFunction > *, BaseLayer< GaussianFunction, arma::mat, arma::mat > *, PositionalEncoding< arma::mat, arma::mat > * > MoreTypes
Linear algebra utility functions, generally performed on matrices or vectors.
Definition: cv.hpp:1