benchmark
20e03326 - Fix bool mask IoU comparison for Detectron2 outputs (#186441)

Commit
48 days ago
Fix bool mask IoU comparison for Detectron2 outputs (#186441) Summary: CPU Inductor freezing can produce tiny numeric differences in Detectron2 mask logits. Those differences are thresholded into boolean pred_masks, so exact bool equality is too strict for these segmentation outputs. The benchmark harness already has an IoU-based bool mask comparator for this class of model, but torch._dynamo.utils.same() dropped use_iou_for_bool and iou_threshold when recursing through Detectron2 Instances objects. As a result, Instances.pred_masks still used exact equality and the affected Mask R-CNN models remained fail_accuracy under freezing. Propagate the bool-mask IoU options through the object and numpy recursive same() paths, enable the existing IoU comparator for the two affected Detectron2 Mask R-CNN models, and update the CPU freezing expected-accuracy files to pass. Fixes #127073 Generated by my agent X-link: https://github.com/pytorch/pytorch/pull/186441 Approved by: https://github.com/eellison, https://github.com/mlazos Reviewed By: wdvr Differential Revision: D108200928 fbshipit-source-id: a54d126c1031cd32ebe5f9bc3f9d7319e2b2f652
Author
Committer
Parents
Loading