So here is another test. I increased the number of workers for dataloading from 1 to 2.  If I use .item() the total time is 3.77 where as if I use .data[0] it is total time 2.71. Previously the time difference was about 6 but now it is about 1. It seems the delay somehow is caused by the dataloading stage which can be reduced by increasing number of workers…