Will you update the code compatible with higher pytorch version, such as cuda11.3, torch1.8?

vasgaowei / D-MIL.pytorch

Codes for: D-MIL: Discrepant multiple instance learning for weakly supervised object detection

https://www.sciencedirect.com/science/article/abs/pii/S0031320321004143

MIT License

15 stars 4 forks source link

Will you update the code compatible with higher pytorch version, such as cuda11.3, torch1.8? #5

Open liz6688 opened 2 years ago

liz6688 commented 2 years ago

Thanks for your work.

I am using Nvidia GPU RTX 3090, which only supports cuda11+. But, higher cuda version only supports higher pytorch version. Did you compile the code on cuda11+? Will you update the code?

Thanks very much.