@vishutanwar and @Shumbabala it may be because the default batch size (64) is too high, so you may need to specify the batch size. Please see this forum post.
If that doesn’t solve it, please share a link to a Google Colab or Kaggle notebook with your code.