WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

Pytorch must be 1.7? #211

Closed Szc-coder closed 2 years ago

Szc-coder commented 2 years ago

In the requirements.txt file,the Pytorch version is 1.7,but my Pytorch version must be 1.8 ,will it affect?

Splendon commented 2 years ago

pytorch1.8 is OK

Szc-coder commented 2 years ago

pytorch1.8 可以

Thank you!