HEVC-CU-depths-prediction-CNN icon indicating copy to clipboard operation
HEVC-CU-depths-prediction-CNN copied to clipboard

Published 20 hours ago • verified publisher icon wolverinn

mapping storage from GPU to the CPU

Open jamesrech opened this issue 7 months ago • 0 comments

I need help, I am trying to reimplement the code on the cpu but I am getting this error: "RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU." I tried to add map_loction =torch.device('cpu') after model.load_state_dict(torch.load(paft.format(LOAD_DIR)) as shown below but I am still getting the same error model.load_state_dict(torch.load(paft.format(LOAD_DIR)),map_location=torch.device('cpu'))

jamesrech avatar Nov 13 '23 22:11 jamesrech