Help regarding running this code on Google Colab

[mahin003]

I have uploded the data set on drive and now i don't understand what to change on ' iam_dataset.py " so that my code works . I did some changes and ran this code on my local computer without any problem but it takes forever on training and kernel dies in the middle of the training and I m totally new on google colab and also not that good on python . If anyone tried this code on google Colab , please share your iam_dataset.py code with me . I've been stuck with it for 4 days now and now i m kind of clueless about what to search for on google so that i can work .
Thats my email : [email protected]