Colaboratory ULMFit.ipynb

I have problem in running “ULMFit.ipynb” notebook at the command line
“learn.fit_one_cycle(1, 2e-2, moms=(0.8,0.7), wd=0.1)” it start to run and after small time my colab disconnect and when I restart the kernal every thing start from scratch.
I have searched this problem and I didn’t find solution with noting that it is public issue.
What can I do?