Stage1: Train 2D pose only

xingyizhou / pytorch-pose-hg-3d

PyTorch implementation for 3D human pose estimation

GNU General Public License v3.0

614 stars 143 forks source link

Open NguyenDangBinh opened 5 years ago

NguyenDangBinh commented 5 years ago

dear, It can be run on GPU, not CPU?

xingyizhou commented 5 years ago

Yes, I haven't tested training with CPU. And it is not recommended to do so since it would be too slow. However if you really want to try, you can change this line https://github.com/xingyizhou/pytorch-pose-hg-3d/blob/master/src/main.py#L38 to opt.device = torch.device('cpu')