SIGSEGV when training on TPU
❓ Questions and Help
Before asking:

Try to find answers to your questions in the Lightning Forum!
Search for similar issues.
Search the docs.


What is your question?
I tried to apply the reformer model on a sentiment analysis task and train it on a tpu. I get a
ProcessExitedException: process X terminated with signal SIGSEGV

What did I do wrong?
Code
   
You can find my code in a colab notebook here.
What have you tried?
I tried to stick to notebook 1 for the general setup and notebook 2 for the tpu setup. I saw #1956 and #2124 however it does not work with the latest version (1.0.4).
What's your environment?

OS: [e.g. iOS, Linux, Win] Linux
Packaging [e.g. pip, conda] pip
Version [e.g. 0.5.2.1] 1.0.4