Any examples on how to use this with Pytorch lightning?

[shengy90]

👋 Hi, are there any examples on how we might be able to use this in conjunction with Pytorch Lightning?

Or would this require us to overwrite the backward method in our LightningModule to use mtl_backward(loss1, loss2) over the default loss.backward()?

[ValerianRey]

Hi! Sorry for the last answer, last week has been crazy in terms of workload.

We didn't test torchjd with Pytorch Lightning yet. Also, I've personally never even used pytorch lightning, so I can't answer you yet.

We'll probably work on this soon, and I agree that adding an example with pytorch lighting integration would be really nice.

[ValerianRey]

#160