Hi,
The basic input image (for a 224x224x3 tensor) would be
2242243=150528 elements (numbers)
602112 bytes = 0.574 Gb
So in the end for a Batch size 2 and having 2 images per input we get
0.57422 = 2.3 Gb
Therefore I think you prob have a bug in the dataset/loader
Soo I would suggest to inspect the shape of the tensor before calling sending the tensor to the gpu as trying to allocate 86 Gb is way too far wrt the size it should be asking.
Another options is that you passed a wrong input to the summary.