We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8d030f9 commit 201cb2aCopy full SHA for 201cb2a
torchvision/models/detection/roi_heads.py
@@ -746,7 +746,7 @@ def forward(
746
if not t["boxes"].dtype in floating_point_types:
747
raise TypeError(f"target boxes must of float type, instead got {t['boxes'].dtype}")
748
if not t["labels"].dtype == torch.int64:
749
- raise TypeError("target labels must of int64 type, instead got {t['labels'].dtype}")
+ raise TypeError(f"target labels must of int64 type, instead got {t['labels'].dtype}")
750
if self.has_keypoint():
751
if not t["keypoints"].dtype == torch.float32:
752
raise TypeError(f"target keypoints must of float type, instead got {t['keypoints'].dtype}")
0 commit comments