While training a neural network I realised the time spent per batch increased when I increased the size of my dataset (without changing the batch size). The important part i