diff --git a/include/cppflow/model.h b/include/cppflow/model.h index acc897c..f52e7f9 100644 --- a/include/cppflow/model.h +++ b/include/cppflow/model.h @@ -52,8 +52,10 @@ namespace cppflow { std::unique_ptr meta_graph = {TF_NewBuffer(), TF_DeleteBuffer}; auto session_deleter = [](TF_Session* sess) { - TF_DeleteSession(sess, context::get_status()); - status_check(context::get_status()); + auto status = TF_NewStatus(); + TF_DeleteSession(sess, status); + status_check(status); + TF_DeleteStatus(status); }; int tag_len = 1;