TFTrainer.train() stucks into infinite loop

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.