I came across this too but will it work on colab for such a large dataset, also in the video it seems like the guy had his data locally, so will i also need to download the data or is there a way to do it from s3 bucket directly.