Performance with different input WFE levels defined during evaluation for models trained on data with a median RMS WFE of 70 nm over 18 modes (blue), and 350 nm RMS over 88 modes (red). Models were also trained on data following a uniform distribution covering the whole input WFE range, using both spatial frequency regimes (cyan and orange). The S/N is 1000 and each training dataset contains 10⁵ samples.