search

shenyunhang / UWSOD

UWSOD: Toward Fully-Supervised-Level Capacity Weakly Supervised Object Detection
https://github.com/shenyunhang/UWSOD/blob/UWSOD/projects/WSL
Apache License 2.0
22 stars 4 forks source link

Can it run on pytorch1.7 cuda11.0 #2

Open Horatio-Ng opened 3 years ago

Horatio-Ng commented 3 years ago

Hi~ I only have an NVIDIA 3090, so can the code run on pytorch1.7 with cuda11.0?

shenyunhang commented 3 years ago

Although I do not test it on pytorch1.7 with cuda11.0, I think it should work. If you have any problem please feel free to open issues.

Horatio-Ng commented 3 years ago

It seems that UWSOD/projects/WSL/setup.py needs to be modified to make it suitable for pytorch1.7. However, I still don't know how to modify it~

shenyunhang commented 3 years ago

I just have installed it with PyTorch 1.7. Can you post the installing log?