Update prototype_source/gpu_direct_storage.py

svekars · web-flow · commit 228614e2becb · 2025-04-28T15:06:24.000-07:00
diff --git a/prototype_source/gpu_direct_storage.py b/prototype_source/gpu_direct_storage.py
@@ -31,7 +31,7 @@
 
 ################################################################################
 # Using GPUDirect Storage with ``torch.save`` and ``torch.load``
-# =============================================================
+# ------------------------------------------------------------------------------------
 # GPUDirect Storage requires a storage alignment of 4KB. You can toggle this by using
 # ``torch.utils.serialization.config.save.storage_alignment``:
 
