Solved by ptrblck in post #2 
              
              
                You can simply modify the forward method of your model and pass an arbitrary number of parameters. 
PyTorch does not use specific “input” tensors etc.