The iteration step for the dataloader takes a long time during training. I tried to increase the number of workers but it doesn't solve the issue