This commit is contained in:
Michael Cusack
2023-08-04 17:22:27 +07:00
parent f8d45f180d
commit 9f8e0857ee
+1 -1
View File
@@ -2,7 +2,7 @@
#!/usr/bin/env python
"""Saves the model as a TorchScript.
The resulting file can be loaded in C++ code and then used for training or infrence with:
The resulting file can be loaded in C++ code and then used for training or inference with:
#include <torch/script.h>
torch::jit::Module module = torch::jit::load("model.pt")
"""