Error in visualize_single_image.py

[devendraswamy]

RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cpu

I tried model = torch.load(model_path,map_location=torch.device('cpu')) and scores, classification, transformed_anchors = model(image.float())

but its not working , please help. Thank you.