When I try to train ALBERT model for SQuAD 2.0 dataset using Tensorflow, TFtrainer.train() gets stuck into infinite loop without any errors or logs.
Below is the link of colab. Any suggestions are welcome.

  
      
      colab.research.google.com
  
  
    

Google Colaboratory 3