Update arguments.py

aws-samples · Oct 9, 2024 · 135f103 · 135f103
1 parent 8677a42
commit 135f103
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/3.test_cases/10.FSDP/model_utils/arguments.py b/3.test_cases/10.FSDP/model_utils/arguments.py
@@ -88,7 +88,7 @@ def parse_args():  # pylint: disable=too-many-statements
     )
     fsdp_grp.add_argument(
         "--cpu_offload",
-        type=int, defalut=0
+        type=int, default=0,
         help="CPU offloading https://pytorch.org/docs/stable/fsdp.html#torch.distributed.fsdp.CPUOffload"
     )