Anamika_Singh:

I am using Colab GPU, is there any limit on size of training data for GPU with 15gb RAM?


Yes, there would always be an upper limit, which fits on your GPU (and of course also system RAM).
You might need to lower the batch size for the particular GPU you are using, if you are running out of memory.