Multi-Task Learning

Hello. My task is to train the FaBSE model for classification and masking tasks. When I train them one by one, I encounter the problem of catastrophic forgetting. Therefore, I want to train the model in parallel. Can you advise me some example of how to do this?