MART
MART copied to clipboard
Remove numpy dependency after upgrading to pytorch 2.x
It looks like the newer PyTorch 2.0 has resolved the issue: https://github.com/pytorch/pytorch/pull/91517#issuecomment-1477998981
Could you add a note in the comment? I think we will upgrade to PyTorch 2.0 soon after torchmetrics
fixes mAP.
Originally posted by @mzweilin in https://github.com/IntelLabs/MART/pull/144#discussion_r1219997665