[atorodius]

good point, I hadn’t realized that you only need to predict chroma! That actully greatly simplifies things

re. chroma subsampling in training data: this is actually a big problem and a good generative model will absolutely learn to predict chroma subsampled values (or JPEG artifacts even!). you can get around it by applying random downscaling with antialiasing during training.