chore:debug tests

mindspore-lab · Jul 22, 2024 · 499a344 · 499a344
1 parent fb1fbc0
commit 499a344
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/modules/test_feature_extraction.py b/tests/modules/test_feature_extraction.py
@@ -205,6 +205,8 @@ def test_feature_extraction_with_checkpoint(model_name, length_target):
         features_only=True,
     )
 
+    print(ms.get_context("device_target"))
+
     assert isinstance(model, nn.Cell), "Loading checkpoint error"
 
     x = ms.Tensor(np.random.randn(8, 3, 32, 32), dtype=ms.float32)