About the TPU category

(Replace this first paragraph with a brief description of your new category. This guidance will appear in the category selection area, so try to keep it below 200 characters.)

Use the following paragraphs for a longer description, or to establish category guidelines or rules:

  • Why should people use this category? What is it for?

  • How exactly is this different than the other categories we already have?

  • What should topics in this category generally contain?

  • Do we need this category? Can we merge with another category, or subcategory?

Hi,

I am facing an issue using Pytorch Lightning for training on TPU. This is about a competition hosted on Drivendata. I am trying a BERT based classification model and trying out TPU training.

However after 1st epoch, the Kernel dies because of Out of Memory error. Request to please guide me. https://www.kaggle.com/aninda/pytorch-lightning-genetic-transformers