@ptrblck thanks , what is the catch of doing this , we are replicating the same decoder layer into different instance variable “self.decoder_recon” and “self.decoder_classifier” and treating one is decoder and other is classifier. Could you please why we need to do this?
self.decoder_recon = nn.Linear(10, 10)
self.decoder_classifier = nn.Linear(10, 10)