本文整理汇总了C++中ModelConfig::sub_models_size方法的典型用法代码示例。如果您正苦于以下问题:C++ ModelConfig::sub_models_size方法的具体用法?C++ ModelConfig::sub_models_size怎么用?C++ ModelConfig::sub_models_size使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类ModelConfig
的用法示例。
在下文中一共展示了ModelConfig::sub_models_size方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: init
void MultiNetwork::init(const ModelConfig& config,
ParamInitCallback callback,
const std::vector<ParameterType>& parameterTypes,
bool useGpu) {
CHECK_GT(config.sub_models_size(), 1) << "sub_models_size should GT 1";
// check submodel[0] is root
CHECK_EQ("root", config.sub_models(0).name())
<< "sub_models(0) should be root";
// ignore root
subNetworks_.resize(config.sub_models_size() - 1);
// base class
NeuralNetwork::init(config, callback, parameterTypes, useGpu);
// sub networks
for (int i = 1; i < config.sub_models_size(); ++i) {
std::string subModelName = config.sub_models(i).name();
if (FLAGS_parallel_nn) {
subNetworks_[i - 1] = std::unique_ptr<ParallelNeuralNetwork>(
new ParallelNeuralNetwork(subModelName, this));
} else {
subNetworks_[i - 1] = std::unique_ptr<NeuralNetwork>(
NeuralNetwork::newNeuralNetwork(subModelName, this));
}
subNetworks_[i - 1]->init(config);
}
}