Revert rnd shadow p from test

mindee · Dec 20, 2024 · 637ac0c · 637ac0c
1 parent 38d5ebe
commit 637ac0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py
@@ -264,7 +264,7 @@ def main(args):
             T.RandomApply(T.GaussianBlur(sigma=(0.5, 1.5)), 0.2),
         ]),
         Compose([
-            T.RandomApply(T.RandomShadow(), 1.0),
+            T.RandomApply(T.RandomShadow(), 0.3),
             T.RandomApply(T.GaussianNoise(), 0.1),
             T.RandomApply(T.GaussianBlur(sigma=(0.5, 1.5)), 0.3),
             RandomGrayscale(p=0.15),