Finding an OMM error while training a ResNet model using GPU

Question

I am Training a ResNet on an NVidia GeForce GTX 1050 ti with 4GB, The image is sized 512x320 using 6 as a batch size which is not really go