fixed tests

vturrisi · Oct 12, 2021 · d020e7e · d020e7e
1 parent aaae112
commit d020e7e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/args/test_setup.py b/tests/args/test_setup.py
@@ -204,6 +204,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [1.0, 0.1],
         "solarization_prob": [0.2, 0.1],
         "min_scale": [0.08],
@@ -234,6 +237,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [0.5],
         "solarization_prob": [0.5],
         "min_scale": [0.08],
@@ -264,6 +270,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [0.5],
         "solarization_prob": [0.5],
         "min_scale": [0.08],
@@ -294,6 +303,9 @@ def test_additional_setup_pretrain():
         "contrast": [0.4],
         "saturation": [0.2],
         "hue": [0.1],
+        "color_jitter_prob": [0.8],
+        "gray_scale_prob": [0.2],
+        "horizontal_flip_prob": [0.5],
         "gaussian_prob": [0.5, 0.2],
         "solarization_prob": [0.5, 0.3],
         "min_scale": [0.08],