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_
Could you add a note in the comment? I think we will upgrade to PyTorch 2.0 soon after
torchmetricsfixes mAP.Originally posted by @mzweilin in #144 (comment)