안녕하세요 yolov5 모델을 GPU상에서 훈련하고 추론할때에 cpu 환경에서 추론시간을 측정해보고 싶은데 방법을 모르겠어서 질문드립니다.

현재 다음과 같이 코드를 작성했는데 cpu 환경에서 돌지 않고 gpu 환경에서 추론을 진행하는 상태입니다.

model = torch.hub.load('./yolov5-master', 'custom', source ='local', path='best.pt',force_reload=True) # 모델 불러오기

model.to(torch.device('cpu'))