@richard  Yes, there’s a reconstruction loss on Y to backprop gradients to the network but not to the c parameter. I’m currently doing the pre-processing inside the network’s forward method but I’d like to have it wrapped in a network layer. Is this possible ?