Training a model with the training dataset pubchem_1 on the TPU V3-8 device took nearly 3 days and 10 hours on average per epoch. Training the same model using a TPU V3-32 pod slice took an average of one day and two hours. Thus, it was decided to train all models on TPU pod slices of V3-32 or higher to speed up the training process.