Thanks for the update. It seems you are now trying to profile the GPU execution time for the posted model (not the data transfer). Since CUDA operations are executed asynchronously, you would have to synchronize the code before starting and stopping the timer via torch.cuda.synchronize(). Did you add these syncs already or are you reusing the previous code?