Update distributed_checkpoint_recipe.rst (#3356)

ShoufaChen · web-flow · commit 67667efbc6ac · 2025-05-08T10:30:46.000-07:00
diff --git a/recipes_source/distributed_checkpoint_recipe.rst b/recipes_source/distributed_checkpoint_recipe.rst
@@ -352,7 +352,7 @@ Alternatively, methods are also provided for users who may wish to convert check
     dcp_to_torch_save(CHECKPOINT_DIR, TORCH_SAVE_CHECKPOINT_DIR)
 
     # converts the torch.save model back to DCP
-    dcp_to_torch_save(TORCH_SAVE_CHECKPOINT_DIR, f"{CHECKPOINT_DIR}_new")
+    torch_save_to_dcp(TORCH_SAVE_CHECKPOINT_DIR, f"{CHECKPOINT_DIR}_new")
 
 
 
