Solved by ptrblck in post #2 
              
              
                You would have to register self.T as an nn.Parameter: 
self.T = nn.Parameter(torch.tensor(guess).unsqueeze(0))

which makes sure it’s returned in model.parameters().