search

THU-DA-6D-Pose-Group / GDR-Net

GDR-Net: Geometry-Guided Direct Regression Network for Monocular 6D Object Pose Estimation. (CVPR 2021)
https://github.com/THU-DA-6D-Pose-Group/GDR-Net
Apache License 2.0
266 stars 46 forks source link

关于pytorch版本的问题 #81

Closed lu-ming-lei closed 2 years ago

lu-ming-lei commented 2 years ago

感谢您更新了新版本的代码。 在旧版本中，我使用配置 pytorch=1.7.0，cuda=10.1，detection=0.4 可以跑通代码 新版中用到了pytorch-lightning，这个需要pytorch版本在1.8.0及以上，我在其他问题中看到您有说pytorch版本在1.8.0及以上跑存在问题，想问一下您的具体pytorch配置 另外想问一下现在版本和您最早发出的版本有何区别 谢谢

wangg12 commented 2 years ago

新版本主要是用了lite让多gpu更丝滑，我之前的经验是pytorch 1.8.x有问题，1.9+没有问题。