From 1571f7e091125e02b03c777a9fe70da8aaea058d Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Tue, 11 Jun 2024 14:20:32 -0700 Subject: [PATCH] enumerate --- tests/trainer/test_fsdp_checkpoint.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/trainer/test_fsdp_checkpoint.py b/tests/trainer/test_fsdp_checkpoint.py index ffa73e42d2..c3924bb372 100644 --- a/tests/trainer/test_fsdp_checkpoint.py +++ b/tests/trainer/test_fsdp_checkpoint.py @@ -790,6 +790,8 @@ def mock_get_checkpoint_validation_function(): pytest.param(4, False, 'adamw', 'amp_bf16', False, None, False, True, False, -1, marks=pytest.mark.world_size(4)), pytest.param(4, False, 'adamw', 'amp_bf16', False, None, False, True, True, 4, marks=pytest.mark.world_size(4)), pytest.param(4, False, 'adamw', 'amp_bf16', False, None, False, True, True, 2, marks=pytest.mark.world_size(4)), + pytest.param(4, False, 'adam', 'amp_bf16', False, None, False, True, True, 2, marks=pytest.mark.world_size(4)), + pytest.param(4, False, 'adamw', 'amp_fp16', False, None, False, True, True, 2, marks=pytest.mark.world_size(4)), ], ) @pytest.mark.filterwarnings(r'ignore:TypedStorage is deprecated.:UserWarning')