Refactor singleton to manage cleanup
This commit is contained in:
@@ -15,9 +15,9 @@ namespace pywrap {
|
||||
torch::Tensor predict(torch::Tensor& X);
|
||||
double score(torch::Tensor& X, torch::Tensor& y);
|
||||
std::string version();
|
||||
std::string graph();
|
||||
std::string callMethodString(const std::string& method);
|
||||
private:
|
||||
PyObject* toPyObject(torch::Tensor& tensor);
|
||||
PyWrap* pyWrap;
|
||||
std::string module;
|
||||
std::string className;
|
||||
|
Reference in New Issue
Block a user