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 df7129b commit def23b7Copy full SHA for def23b7
tests/models/test_manet.py
@@ -1,6 +1,27 @@
1
+import segmentation_models_pytorch as smp
2
+
3
from tests.models import base
4
5
6
class TestManetModel(base.BaseModelTester):
7
test_model_type = "manet"
8
files_for_diff = [r"decoders/manet/", r"base/"]
9
10
+ def test_interpolation(self):
11
+ # test bilinear
12
+ model_1 = smp.create_model(
13
+ self.test_model_type,
14
+ self.test_encoder_name,
15
+ decoder_interpolation="bilinear",
16
+ )
17
+ for block in model_1.decoder.blocks:
18
+ assert block.interpolation_mode == "bilinear"
19
20
+ # test bicubic
21
+ model_2 = smp.create_model(
22
23
24
+ decoder_interpolation="bicubic",
25
26
+ for block in model_2.decoder.blocks:
27
+ assert block.interpolation_mode == "bicubic"
0 commit comments