Tensorflow会话内存泄漏

时间:2017-05-09 15:05:08

标签: session memory-leaks tensorflow objective-c++

我正在使用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会话吗?

0 个答案:

没有答案