Browse Source

Update requirements.txt

Bobholamovic 3 years ago
parent
commit
9e6e6c717c
1 changed files with 4 additions and 0 deletions
  1. 4 0
      tests/rs_models/test_seg_models.py

+ 4 - 0
tests/rs_models/test_seg_models.py

@@ -52,3 +52,7 @@ class TestFarSegModel(TestSegModel):
         self.specs = [
             dict(), dict(num_classes=20), dict(encoder_pretrained=False)
         ]
+
+    def set_targets(self):
+        self.targets = [[self.get_zeros_array(16)], [self.get_zeros_array(20)],
+                        [self.get_zeros_array(16)]]