Solved by albanD in post #2 
              
              
                Hi, 
torch.cuda.synchronize() just wait for all the work to be done on the GPU. 
If the synchronize takes different time for different models, I guess that means that one requires more things to be done on the GPU. There is not much you can do here. 
Also, unless you are doing time meansurement, you…