Skip to content

Conversation

@sridhs21
Copy link
Contributor

Reduced augmentation aggressiveness to prevent underfitting:

  • Rotation probability: 75% → 50% (line 346)
    • Less frequent random rotations (90°, 180°, 270°).
  • Gaussian noise probability: 30% → 10% (line 360)
    • Significantly reduced noise injection to preserve X-point magnetic field structure.
  • Cutout/Random erasing probability: 20% → 5% (line 373)
    • Much less frequent random erasing to avoid destroying small X-point features.

Parameters to Test with for now:

--learningRate 5e-4 \
--weightDecay 5e-5 \
--dropoutRate 0.15 \
--batchSize 64 \
--patience 100 \
--minTrainingLoss 0

@cwsmith
Copy link
Contributor

cwsmith commented Nov 19, 2025

Thank you. The job was just submitted.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants