Unverified Commit 344ccc05 authored by Robert Perrotta's avatar Robert Perrotta Committed by GitHub
Browse files

Fix missing f-string prefix in error message (#6684)


Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent e89f639f
......@@ -746,7 +746,7 @@ class RoIHeads(nn.Module):
if not t["boxes"].dtype in floating_point_types:
raise TypeError(f"target boxes must of float type, instead got {t['boxes'].dtype}")
if not t["labels"].dtype == torch.int64:
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}")
if self.has_keypoint():
if not t["keypoints"].dtype == torch.float32:
raise TypeError(f"target keypoints must of float type, instead got {t['keypoints'].dtype}")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment