From ac2b435151a644dbdb0544a623febba905623085 Mon Sep 17 00:00:00 2001 From: Michael Cusack Date: Fri, 4 Aug 2023 16:55:26 +0700 Subject: [PATCH] docs --- save_model.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/save_model.py b/save_model.py index 54bf82a..c80253a 100644 --- a/save_model.py +++ b/save_model.py @@ -1,6 +1,11 @@ #!/usr/bin/env python3 #!/usr/bin/env python -"""Saves the model as a TorchScript.""" +"""Saves the model as a TorchScript. + +The resulting file can be loaded in C++ code and then used for training or infrence with: + #include + torch::jit::Module module = torch::jit::load("model.pt") +""" import glob import os