Solved by ptrblck in post #2 
              
              
                Yes, it would be possible to split the forward passes and call them separately: 
# original forward
def forward(self, x):
    out = self.forward_encoder(x)
    out = self.forward_decoder(out)
    return out

and later call model.forward_decoder(input) to call the decoder only. 
However, since you ar…