Ejemplos de Net::GetWeights en C++ (Cpp)

Ejemplo n.º 1

Archivo: genweights_mex.cpp Proyecto: 1165048017/ConvNet

void mexFunction(int nLhs, mxArray* pLhs[], int nRhs, const mxArray* pRhs[]) {

  mexAssert(nRhs == NARGIN, "Number of input arguments in wrong!");
  mexAssert(nLhs == NARGOUT, "Number of output arguments is wrong!" );
  
  Net net;
  net.InitLayers(IN_L);
  net.InitParams(IN_P);
  net.InitWeights(NULL);
  net.GetWeights(OUT_W);
}

Ejemplo n.º 2

Archivo: genweights_mex.cpp Proyecto: yuanpu/ConvNet

void mexFunction(int nLhs, mxArray* pLhs[], int nRhs, const mxArray* pRhs[]) {

  mexAssert(nRhs == NARGIN, "Number of input arguments in wrong!");
  mexAssert(nLhs == NARGOUT, "Number of output arguments is wrong!" );
  
  size_t seed = (size_t) mexGetScalar(IN_S);
  
  Net net;
  net.InitRand(seed);
  net.InitLayers(IN_L);  
  net.InitWeights(NULL);
  net.GetWeights(OUT_W);
}