Skip to content

GitLab

Unverified Commit fe83261b authored by Cao Yuhang's avatar Cao Yuhang Committed by GitHub
Browse files

revert dcn (#597)

parent f3a2be99
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
mmcv/ops/csrc/pytorch/deform_conv_cuda.cu
View file @ fe83261b
...@@ -278,8 +278,6 @@ void DeformConvForwardCUDAKernelLauncher(Tensor input, Tensor weight, ...@@ -278,8 +278,6 @@ void DeformConvForwardCUDAKernelLauncher(Tensor input, Tensor weight,
} }
columns = columns =
columns.view({columns.size(0) * columns.size(1), columns.size(2)}); columns.view({columns.size(0) * columns.size(1), columns.size(2)});
weight = weight.view({weight.size(0) * weight.size(1), weight.size(2),
weight.size(3), weight.size(4)});
} }
output_buffer = output_buffer.view( output_buffer = output_buffer.view(
...@@ -377,7 +375,6 @@ void DeformConvBackwardInputCUDAKernelLauncher( ...@@ -377,7 +375,6 @@ void DeformConvBackwardInputCUDAKernelLauncher(
gradOutput = gradOutput.view( gradOutput = gradOutput.view(
{gradOutput.size(0), gradOutput.size(1) * gradOutput.size(2), {gradOutput.size(0), gradOutput.size(1) * gradOutput.size(2),
gradOutput.size(3), gradOutput.size(4), gradOutput.size(5)}); gradOutput.size(3), gradOutput.size(4), gradOutput.size(5)});
weight = weight.view({nOutputPlane, nInputPlane, kH, kW});
deformable_col2im_coord(columns, input[elt], offset[elt], nInputPlane, deformable_col2im_coord(columns, input[elt], offset[elt], nInputPlane,
inputHeight, inputWidth, kH, kW, padH, padW, dH, dW, inputHeight, inputWidth, kH, kW, padH, padW, dH, dW,
......
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