search

mkocabas / VIBE

Official implementation of CVPR2020 paper "VIBE: Video Inference for Human Body Pose and Shape Estimation"
https://arxiv.org/abs/1912.05656
Other
2.86k stars 550 forks source link

[BUG]CPU without GPU #219

Closed zhangchengzhi-max closed 2 years ago

zhangchengzhi-max commented 2 years ago

The last step is this error. My computer has no GPU. According to this prompt, I don't know where to change the code. Please help. Thank you 132863565-4868ff14-1d2c-4286-8061-ac231edd3270

liuhaorandezhanghao commented 2 years ago

device改成device=torch.device('cpu'),还有ckpt = torch.load(pretrained_file,map_location='cpu')