Skip to content

GitLab

Unverified Commit a2d240ee authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Update the error message on DeformConv2d. (#4908)

parent 272e080c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/csrc/ops/cpu/deform_conv2d_kernel.cpp
View file @ a2d240ee
...@@ -963,7 +963,7 @@ at::Tensor deform_conv2d_forward_kernel( ...@@ -963,7 +963,7 @@ at::Tensor deform_conv2d_forward_kernel(
(mask_c.size(0) == input_c.size(0)), "invalid batch size of mask"); (mask_c.size(0) == input_c.size(0)), "invalid batch size of mask");
TORCH_CHECK( TORCH_CHECK(
(!use_mask || (mask_c.size(2) == out_h && mask_c.size(3) == out_w)), (!use_mask || (mask_c.size(2) == out_h && mask_c.size(3) == out_w)),
"offset output dims: (", "mask output dims: (",
mask_c.size(2), mask_c.size(2),
", ", ", ",
mask_c.size(3), mask_c.size(3),
......
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