Hi, I solved this problem through upload the data to the temporary folder in colab instead read the data from the google drive which is extremely slow in the first epoch. Below is the difference between read from temporary folder(/content/)and google drive.
WeChat2832ca0ee811b9bffc14a3eca28c343e618×606 236 KB