DMLC　Parameter　例子学习

最新推荐文章于 2024-09-21 22:34:33 发布

原创最新推荐文章于 2024-09-21 22:34:33 发布 · 2.8k 阅读

0 ·

CC 4.0 BY-SA版权

DMLC 专栏收录该内容

1 篇文章

订阅专栏

本文介绍如何在DMLC库中配置参数，通过具体的代码示例展示了如何定义参数结构体、注册参数以及初始化参数的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DMLC: Distributed (Deep) Machine Learning Community
是一种库，这里面学习的是其操作（类似于caffe里面的层）里面参数设置，通过继承dmlc里面的Parameter这个类，使用dmlc里面的一些宏定义来实现这个功能，具体例子可以参见下面的例子。

#include<dmlc/parameter.h>
#include<iostream>

// declare the parameter, normally put it in header file.
struct MyParam : public dmlc::Parameter<MyParam> {
  float learning_rate;
  int num_hidden;
  int activation;
  std::string name;
  // declare parameters
  DMLC_DECLARE_PARAMETER(MyParam) {
    DMLC_DECLARE_FIELD(num_hidden).set_range(0, 1000)
        .describe("Number of hidden unit in the fully connected layer.");
    DMLC_DECLARE_FIELD(learning_rate).set_default(0.01f)
        .describe("Learning rate of SGD optimization.");
    DMLC_DECLARE_FIELD(activation).add_enum("relu", 1).add_enum("sigmoid", 2)
        .describe("Activation function type.");
    DMLC_DECLARE_FIELD(name).set_default("layer")
        .describe("Name of the net.");
  }
};

// register the parameter, this is normally in a cc file.
DMLC_REGISTER_PARAMETER(MyParam);

int main() {
   MyParam param;
   std::vector<std::pair<std::string, std::string> > param_data = {
     {"num_hidden", "100"},
     {"activation", "relu"},
     {"name", "myname"}
   };
   // set the parameters
   param.Init(param_data);
   std::cout << param.name << std::endl;
   param.name="hello";
   std::cout << param.name << std::endl;
   return 0;
}