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 3c2c002 commit 9ed2fa3Copy full SHA for 9ed2fa3
torchvision/csrc/cpu/nms_cpu.cpp
@@ -6,9 +6,11 @@ at::Tensor nms_cpu_kernel(
6
const at::Tensor& scores,
7
const float iou_threshold) {
8
AT_ASSERTM(!dets.options().device().is_cuda(), "dets must be a CPU tensor");
9
- AT_ASSERTM(!scores.options().device().is_cuda(), "scores must be a CPU tensor");
10
AT_ASSERTM(
11
- dets.scalar_type() == scores.scalar_type(), "dets should have the same type as scores");
+ !scores.options().device().is_cuda(), "scores must be a CPU tensor");
+ AT_ASSERTM(
12
+ dets.scalar_type() == scores.scalar_type(),
13
+ "dets should have the same type as scores");
14
15
if (dets.numel() == 0)
16
return at::empty({0}, dets.options().dtype(at::kLong));
0 commit comments