i’m concatenating two different branches of my model and after, it goes through an nn.Linear layer with a relu activation.
Do i need to pass gradients?
I’m computing the BCELoss and then calling loss.backward()