Makes sense, especially if you then want to just call the output of the encoder after training to get the compressed data…so then you don’t have to write a separate model you can just access the training model in evaluation mode and ignore the decoder.