We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c2064f6 commit 313b9f9Copy full SHA for 313b9f9
tests/test_backbones.py
@@ -44,7 +44,7 @@ def test_can_create_backbones_from_timm_features_only(model_name, test_input, re
44
backbone(input_tensor)
45
46
47
-@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300", "prithvi_swin_B"])
+@pytest.mark.parametrize("model_name", ["prithvi_vit_100", "prithvi_vit_300"])
48
def test_vit_models_accept_multitemporal(model_name, input_224_multitemporal):
49
backbone = timm.create_model(model_name, pretrained=False, num_frames=NUM_FRAMES)
50
backbone(input_224_multitemporal)
0 commit comments