Parameters of all three models are fed into the optimizer, the trained model is actually a “LossWrapper” instance. How my code works is generally the same as in the link: https://github.com/Hui-Li/multi-task-learning-example-PyTorch/blob/master/multi-task-learning-example-PyTorch.ipynb 5