Hi!  This probably will not answer your question completely. If the dataset is small I would move it to my local file system in colab that is ‘/content/’, this is because in colab there is a cap on number of times you can load the data from drive to bypass it it is recommended to put the data in local file as there is no cap on number times you load the data from  local system.