How to perform training with two tasks?

[hyz026]

Thank you for your work!

In your code
https://github.com/chenllliang/DreamEngine/blob/main/src/scripts/train/train.py#L1044
The model can only be trained with 1 task per iteration.

However, in your paper, stage 1 and stage 2 are trained with two tasks. Can you provide an example of your original paper?