search

NVlabs / Deep_Object_Pose

Deep Object Pose Estimation (DOPE) – ROS inference (CoRL 2018)
Other
1.03k stars 287 forks source link

Train problem with train.py in script/train2 #310

Open huyle24022002 opened 1 year ago

huyle24022002 commented 1 year ago

Hi, I have installed all requirement package in requirement, i also install and upgrade cuda but i still not run train.py file, there is my error. Thank you to any support <3 error1 error2

TontonTremblay commented 1 year ago

Which version of python and torch?

huyle24022002 commented 1 year ago

Which version of python and torch Python 3.8 and Pytorch 2.0.1+cu117

TontonTremblay commented 1 year ago

Can you try with pytorch 1.8? I think they changed how things worked in 2.0 and we would need to update the code to work with it, I welcome PR if you end up doing it !

huyle24022002 commented 1 year ago

Can you try with pytorch 1.8? I think they changed how things worked in 2.0 and we would need to update the code to work with it, I welcome PR if you end up doing it !

I will try it now and response soon, thank you very much