我正在使用tensorlfow C ++ API,我已经链接张量流框架以使用inference.pb文件执行预测。
推理有效但我有一个内存泄漏(根据工具)链接到session-> Close()。
我的功能是:
b'\x04\x0e\x1d'
//与加载和创建会话相关的其他功能是:
- (double) MLPredictionObjC: (float[24]) inputFeatures {
double results[]={0.0};
NSString *path = [[NSBundle mainBundle] pathForResource:@"inference84" ofType:@"pb"];
if ([self loadGraphFromPath:path] && [self createSession]) {
double resultsP = [self predictML:inputFeatures];
session->Close();
return resultsP;
}
return *results;
}
我不知道如何解决这个问题?我应该使用std :: unique_ptr会话吗?