I am trying to train a Regression model for a series of images on my Local PC running on an Nvidia RTX 2070 Super.
Input image Shape: (96,160,3), 160