Tensorflow::Session 释放内存
生活随笔
收集整理的這篇文章主要介紹了
Tensorflow::Session 释放内存
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
使用tensorflow::Session是要釋放的,否則循環起來,崩潰到懷疑人生。
方法一:
使用Close()
tensorflow::Session * session;tensorflow::Status status; tensorflow::GraphDef graphdef; //Define a graph for the current modelstatus = tensorflow::NewSession(tensorflow::SessionOptions(), &session); // Create a new session Sessiontensorflow::Status status_load = tensorflow::ReadBinaryProto(tensorflow::Env::Default(), model_path, &graphdef); // read graph model from pb filetensorflow::Status status_create = session->Create(graphdef); // Import the model into the session Session status_run = session->Run(...);status = session->Close();delete session;
方法二:
使用智能指針
tensorflow::SessionOptions options;tensorflow::Status status; tensorflow::GraphDef graphdef; //Define a graph for the current modelstd::unique_ptr<tensorflow::Session> session = std::unique_ptr<tensorflow::Session>(tensorflow::NewSession(options)); // Create a new session Sessiontensorflow::Status status_load = tensorflow::ReadBinaryProto(tensorflow::Env::Default(), model_path, &graphdef); // read graph model from pb filetensorflow::Status status_create = session->Create(graphdef); // Import the model into the session Session status_run = session->Run(...);
總結
以上是生活随笔為你收集整理的Tensorflow::Session 释放内存的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 基于CUDA实现立方体贴图 (Cubem
- 下一篇: 图像的矩,以及利用矩求图像的重心,方向