Update save sizes into GiB (pytorch#2143)

sarvamai · Dec 23, 2024 · 68ddd09 · 68ddd09
1 parent 6bad322
commit 68ddd09
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchtune/training/checkpointing/_checkpointer.py b/torchtune/training/checkpointing/_checkpointer.py
@@ -300,7 +300,7 @@ def save_checkpoint(
             torch.save(state_dict[training.MODEL_KEY], output_path)
             logger.info(
                 "Model checkpoint of size "
-                f"{os.path.getsize(output_path) / 1000**3:.2f} GB "
+                f"{os.path.getsize(output_path) / 1024**3:.2f} GiB "
                 f"saved to {output_path}"
             )
 
@@ -819,7 +819,7 @@ def save_checkpoint(
                     )
                 logger.info(
                     "Adapter checkpoint of size "
-                    f"{os.path.getsize(output_path) / 1000**3:.2f} GB "
+                    f"{os.path.getsize(output_path) / 1024**3:.2f} GiB "
                     f"saved to {output_path}"
                 )
         elif adapter_only: