From 135f103ed5e3a052c9ad3a7988ddf55515c68e54 Mon Sep 17 00:00:00 2001 From: Keita Watanabe Date: Thu, 10 Oct 2024 07:43:11 +0900 Subject: [PATCH] Update arguments.py --- 3.test_cases/10.FSDP/model_utils/arguments.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/3.test_cases/10.FSDP/model_utils/arguments.py b/3.test_cases/10.FSDP/model_utils/arguments.py index 0504246d..a988d83b 100644 --- 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" )