Skip to content

Commit 28745d4

Browse files
greatly reduce the augmentations for the training
1 parent 078d4aa commit 28745d4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/train.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,8 +225,8 @@ def main(args):
225225
timesteps=timesteps,
226226
stepsSampling='uniform',
227227
# no augmentations by default
228-
pointsNoise=0.02, pointsDropout=0.1,
229-
eyesDropout=0.1, eyesAdditiveNoise=0.01, brightnessFactor=5.0, lightBlobFactor=5.0,
228+
pointsNoise=0.01, pointsDropout=0.0,
229+
eyesDropout=0.1, eyesAdditiveNoise=0.01, brightnessFactor=1.5, lightBlobFactor=1.5,
230230
),
231231
)
232232
)

0 commit comments

Comments
 (0)