I wasted several hours until I discovered that reducing the batch size and resizing the width of my input image (image size) were necessary steps.
I wasted several hours until I discovered that reducing the batch size and resizing the width of my input image (image size) were necessary steps.