I have a network, that takes two augmented images (xi,xj) of the same sample x. I stack the two images in one input in the getitem function.
Now, in my training function, how can I unstack the images (xi,xj) for each input in each batch, before feeding them to the model?
Thank you so much!