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

support Pytorch >=1.5 #4

Closed bityangke closed 3 years ago

bityangke commented 3 years ago

Hi Wei, How can D-MIL support Pytorch above 1.4?

vasgaowei commented 3 years ago

It's sorry that only pytorch 1.0 - 1.4 is supported now. And branch supports pytorch above 1.4.

bityangke commented 3 years ago

Thanks!