Train model on Colab TPU with distributed strategy

Question

I\'m trying to train and run an image classification model on Colab, using TPU. No pytorch.

I know that TPU works only with files from GCS buckets so I load the datas