WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.69k stars 307 forks source link

Fix compilation issue on PyTorch > 1.4 #161

Closed seonho closed 3 years ago

seonho commented 3 years ago

151

danielcrane commented 3 years ago

This was a life saver for me, thanks! Hope it can get merged eventually.

WXinlong commented 3 years ago

Thank you for the PR!