Skip to content

Commit 50bcf7c

Browse files
committed
Add test
1 parent 3b86e89 commit 50bcf7c

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

tests/encoders/test_common.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)