Performance with altered S/N levels during evaluation for three models trained on data with a median RMS WFE of 70 nm over 18 modes, with an S/N of 30 (purple), 1000 (blue), and with S/Ns uniformly distributed over the entire S/N range (green). Each point is obtained from an testing batch composed of 1000 samples (the median value together with the 2–98th percentiles are shown).