Both types of models (C and H) were trained and evaluated in the same way as in our previous work (Ref. ). Specifically, to train a model, the following loss function (Eq.11 defined in Ref. ) was minimized via stochastic gradient descent (using Adam optimizer), with respect to the set of model parameters (Θ; located inside the coupling layers; blue in Fig. ):