Pytorch_Realtime_Multi-Person_Pose_Estimation icon indicating copy to clipboard operation
Pytorch_Realtime_Multi-Person_Pose_Estimation copied to clipboard

Published 20 hours ago • verified publisher icon last-one

one gpu

Open g5996706 opened this issue 6 years ago • 2 comments

Hi, If I only have one gpu ,how can I run it? thank you for your answer

g5996706 avatar Dec 01 '17 13:12 g5996706

change the line 49 "model = torch.nn.DataParallel(model, device_ids=args.gpu).cuda()" in train_pose.py to "model = model.cuda()"

last-one avatar Dec 01 '17 15:12 last-one

Thanks you very much!!!

g5996706 avatar Dec 03 '17 11:12 g5996706