greatly reduce the augmentations for the training

GreenWizard2015 · Jun 23, 2024 · 28745d4 · 28745d4
1 parent 078d4aa
commit 28745d4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scripts/train.py b/scripts/train.py
@@ -225,8 +225,8 @@ def main(args):
         timesteps=timesteps,
         stepsSampling='uniform',
         # no augmentations by default
-        pointsNoise=0.02, pointsDropout=0.1,
-        eyesDropout=0.1, eyesAdditiveNoise=0.01, brightnessFactor=5.0, lightBlobFactor=5.0,
+        pointsNoise=0.01, pointsDropout=0.0,
+        eyesDropout=0.1, eyesAdditiveNoise=0.01, brightnessFactor=1.5, lightBlobFactor=1.5,
       ),
     )
   )