소스 검색

Update FarSeg unittest

Bobholamovic 2 년 전
부모
커밋
e0497a0ceb
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      tests/rs_models/test_seg_models.py

+ 1 - 1
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)
         ]