After using torch.cuda.synchronize before time, the output of time is given as follow:
Any suggestion in this regard?
torch.cuda.synchronize()
start.record()
for i,[batch_image, batch_id_label, batch_pose_label] in enumerate(dataloader):
        end.record()
        
        print("time of or loop is ", start.elapsed_time(end))


image.png1643×335 23.4 KB