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/script.h>
+    torch::jit::Module module = torch::jit::load("model.pt")
+"""
 
 import glob
 import os