Keep import style as saving

angerrp · angerrp · commit ab10ae5170a8 · 2024-07-22T10:58:30.000+02:00
diff --git a/recipes_source/distributed_checkpoint_recipe.rst b/recipes_source/distributed_checkpoint_recipe.rst
@@ -237,7 +237,7 @@ the intent is to save or load in "non-distributed" style, meaning entirely in th
     import os
 
     import torch
-    import torch.distributed.checkpoint as DCP
+    import torch.distributed.checkpoint as dcp
     import torch.nn as nn
 
 
@@ -263,7 +263,7 @@ the intent is to save or load in "non-distributed" style, meaning entirely in th
         }
 
         # since no progress group is initialized, DCP will disable any collectives.
-        DCP.load(
+        dcp.load(
             state_dict=state_dict,
             checkpoint_id=CHECKPOINT_DIR,
         )

Original file line number	Diff line number	Diff line change
`@@ -237,7 +237,7 @@ the intent is to save or load in "non-distributed" style, meaning entirely in th`
`237`	`237`	`import os`
`238`	`238`
`239`	`239`	`import torch`
`240`		`- import torch.distributed.checkpoint as DCP`
	`240`	`+ import torch.distributed.checkpoint as dcp`
`241`	`241`	`import torch.nn as nn`
`242`	`242`
`243`	`243`
`@@ -263,7 +263,7 @@ the intent is to save or load in "non-distributed" style, meaning entirely in th`
`263`	`263`	`}`
`264`	`264`
`265`	`265`	`# since no progress group is initialized, DCP will disable any collectives.`
`266`		`- DCP.load(`
	`266`	`+ dcp.load(`
`267`	`267`	`state_dict=state_dict,`
`268`	`268`	`checkpoint_id=CHECKPOINT_DIR,`
`269`	`269`	`)`