When testing the effect of autoencoder roundtrip errors in online prognostic runs, some runs crash due to specific humidity going negative. This PR adds the output_limiter_config option to the DenseAutoencoder, where the final layer in the model limits the output range. Similar to the dense model training, the loss is calculated on the limited outputs.
When testing the effect of autoencoder roundtrip errors in online prognostic runs, some runs crash due to specific humidity going negative. This PR adds the
output_limiter_configoption to theDenseAutoencoder, where the final layer in the model limits the output range. Similar to the dense model training, the loss is calculated on the limited outputs.Added public API:
DenseAutoencoderHyperparameters.output_limit_configCoverage reports (updated automatically):