enumerate

mosaicml · Jun 11, 2024 · 1571f7e · 1571f7e
1 parent e65e4fd
commit 1571f7e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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')