Add Checkpointer Tutorial + Docs (#674)

pytorch · Apr 11, 2024 · 0539976 · 0539976
1 parent a899c7c
commit 0539976
Show file tree

Hide file tree

Showing 10 changed files with 591 additions and 207 deletions.
diff --git a/docs/source/api_ref_utilities.rst b/docs/source/api_ref_utilities.rst
@@ -4,6 +4,19 @@ torchtune.utils
 
 .. currentmodule:: torchtune.utils
 
+
+.. _checkpointing_label:
+
+Checkpointing
+-------------
+
+.. autosummary::
+    :toctree: generated/
+    :nosignatures:
+
+    FullModelHFCheckpointer
+    FullModelMetaCheckpointer
+
 .. _dist_label:
 
 Distributed