Sorry for the confusion. In fact, I just want to know how to do BP in multi-task learning in pytorch.
Just forget about the GAN, let’s say the architecture is:
block A - > block B → loss_B
|_____> block C - >loss_C
the model parameters include block A, B and C. optimizer is torch.optim.Adam(model.parameters()), and total loss = loss_B + loss_C. Is this correct to train the multi-task learning model?
Thank you.