search

ZhengdiYu / Arbitrary-Hands-3D-Reconstruction

🔥(CVPR 2023) ACR: Attention Collaboration-based Regressor for Arbitrary Two-Hand Reconstruction
https://semanticdh.github.io/ACR/
MIT License
174 stars 12 forks source link

parallel issue #29

Closed sora158 closed 1 month ago

sora158 commented 1 month ago

some version of pytorch may have parallel issues, just fixed like below:

Line61 in acr.main

self.model = model.cuda() #nn.DataParallel(model.cuda())