Thanks for your reply. I am measuring the time for registration process (for loop) and data loading step is excluded. I have added torch.cuda.synchronize() before starting and stopping the timer and the difference still exists.