Update FarSeg unittest

2 years ago · e0497a0ceb
parent 0b87782eec
commit e0497a0ceb
1 changed files with 1 additions and 1 deletions
--- a/tests/rs_models/test_seg_models.py
+++ b/tests/rs_models/test_seg_models.py
@ -50,7 +50,7 @@ class TestFarSegModel(TestSegModel):

    def set_specs(self):
        self.specs = [
-            dict(), dict(num_classes=20), dict(encoder_pretrained=False),
+            dict(), dict(num_classes=20), dict(pretrained_encoder=False),
            dict(in_channels=10)
        ]