Thanks for your reply, I suppose you meant warmup iterations? But I got the general idea to take mean over number of iterations.
Also it prints this in the end:

This might be silly question but please bear with me as I am curious here;  the cuda time corresponds to the time taken to compute the prediction
pred = model(x)

What exactly does the CPU time corresponds here?