feat(doc): updated model export

jolibrain · Jun 21, 2023 · e692f78 · e692f78
1 parent b5bf121
commit e692f78
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/docs/source/export.rst b/docs/source/export.rst
@@ -0,0 +1,15 @@
+#############
+ Model export
+############
+
+************
+Model export
+************
+
+.. code:: bash
+
+   python3 -m scripts.export_jit_model --model-in-file "/path/to/model_checkpoint.pth" --model-out-file exported_model.pt --model-type mobile_resnet_9blocks --img-size 360
+
+Then ``exported_model.pt`` can be reloaded without JoliGEN to perform
+inference with an external software, e.g. `DeepDetect
+<https://github.com/jolibrain/deepdetect>`_ with torch backend.
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -174,8 +174,9 @@ program `"Confiance.AI" <https://www.confiance.ai/en/>`_
    :glob:
    :hidden:
 
+   export
    inference
-
+   
 .. toctree::
    :maxdepth: 2
    :caption: Server & API