We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3b86e89 commit 50bcf7cCopy full SHA for 50bcf7c
tests/encoders/test_common.py
@@ -0,0 +1,15 @@
1
+import pytest
2
+import segmentation_models_pytorch as smp
3
+from tests.utils import slow_test
4
+
5
6
+@pytest.mark.parametrize(
7
+ "encoder_name_and_weights",
8
+ [
9
+ ("resnet18", "imagenet"),
10
+ ],
11
+)
12
+@slow_test
13
+def test_load_encoder_from_hub(encoder_name_and_weights):
14
+ encoder_name, weights = encoder_name_and_weights
15
+ smp.encoders.get_encoder(encoder_name, weights=weights)
0 commit comments