@Mdrnfox Hmm, I’m still confused as to how the two dataloders would be taken into account. For instance, the compute_loss function takes in a single parameter as input. I assume that this must be overridden to have two input parameters?
Moreover, that would mean that the function that calls the compute_loss function (training_step()) would have to take two inputs as well, no?