本文整理汇总了C++中Solver::Snapshot方法的典型用法代码示例。如果您正苦于以下问题:C++ Solver::Snapshot方法的具体用法?C++ Solver::Snapshot怎么用?C++ Solver::Snapshot使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类Solver
的用法示例。
在下文中一共展示了Solver::Snapshot方法的1个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的C++代码示例。
示例1: solver_snapshot
// ====================================================================================
// Usage: caffe_('solver_snapshot', hSolver, save_file)
//20160215find
static void solver_snapshot(MEX_ARGS) {
mxCHECK(nrhs == 2 && mxIsStruct(prhs[0]) && mxIsChar(prhs[1]),
"Usage: caffe_('solver_snapshot', hSolver, save_file)");
Solver<float>* solver = handle_to_ptr<Solver<float> >(prhs[0]);
char* snapshot_file = mxArrayToString(prhs[1]);
if (snapshot_file[0] == '\0') {
solver->Snapshot();
}
else {
const string *prefix = &(solver->param().snapshot_prefix());
string ori_prefix = *prefix;
*(string*)prefix = "@" + (string)snapshot_file;
solver->Snapshot();
*(string*)prefix = ori_prefix;
mxFree(snapshot_file);
}
}