Thank you so much. However I have a slowdown not only at the beginning of every epoch but also in each cycle of for sample in train_data. I think this slowdown is caused by the data transformation (rotation and zooming) inside the__getitem__ method of my custom dataset. How I can perform in parallel data loading and training?