本文整理汇总了C++中NeuralNetwork::exportLayers方法的典型用法代码示例。如果您正苦于以下问题:C++ NeuralNetwork::exportLayers方法的具体用法?C++ NeuralNetwork::exportLayers怎么用?C++ NeuralNetwork::exportLayers使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类NeuralNetwork
的用法示例。
在下文中一共展示了NeuralNetwork::exportLayers方法的2个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: saveNetwork
void saveNetwork(const NeuralNetwork<TDevice> &nn, const std::string &filename)
{
rapidjson::Document jsonDoc;
jsonDoc.SetObject();
nn.exportLayers (&jsonDoc);
nn.exportWeights(&jsonDoc);
FILE *file = fopen(filename.c_str(), "w");
if (!file)
throw std::runtime_error("Cannot open file");
rapidjson::FileStream os(file);
rapidjson::PrettyWriter<rapidjson::FileStream> writer(os);
jsonDoc.Accept(writer);
fclose(file);
}
示例2: saveState
void saveState(const NeuralNetwork<TDevice> &nn, const optimizers::lmOptimizer<TDevice> &optimizer, const std::string &infoRows)
{
// create the JSON document
rapidjson::Document jsonDoc;
jsonDoc.SetObject();
// add the configuration options
jsonDoc.AddMember("configuration", Configuration::instance().serializedOptions().c_str(), jsonDoc.GetAllocator());
// add the info rows
std::string tmp = boost::replace_all_copy(infoRows, "\n", ";;;");
jsonDoc.AddMember("info_rows", tmp.c_str(), jsonDoc.GetAllocator());
// add the network structure and weights
nn.exportLayers (&jsonDoc);
nn.exportWeights(&jsonDoc);
// add the state of the optimizer
optimizer.exportState(&jsonDoc);
// open the file
std::stringstream autosaveFilename;
std::string prefix = Configuration::instance().autosavePrefix();
autosaveFilename << prefix;
if (!prefix.empty())
autosaveFilename << '_';
autosaveFilename << "epoch";
autosaveFilename << std::setfill('0') << std::setw(3) << optimizer.currentEpoch();
autosaveFilename << ".autosave";
std::string autosaveFilename_str = autosaveFilename.str();
FILE *file = fopen(autosaveFilename_str.c_str(), "w");
if (!file)
throw std::runtime_error("Cannot open file");
// write the file
rapidjson::FileStream os(file);
rapidjson::PrettyWriter<rapidjson::FileStream> writer(os);
jsonDoc.Accept(writer);
fclose(file);
}