Potential device freezing while training encountered. Need solutions...

It has been seen during the run of histroy : in amazon_reviews_final.ipynb the device freezes. This is a huge problem. One solution is we train the model in Cloud instead of Local server or Host Machine. Any new leads or ideas?