Imported from https://github.com/Hui-Li/multi-task-learning-example-PyTorch