"docs/vscode:/vscode.git/clone" did not exist on "c14cfb10362e07db99bded850a013bdf2522bb7e"
Commit 170b66f5 authored by yhcao6's avatar yhcao6
Browse files

fix bug of deform_conv_backward_input_cuda in deform_conv_cuda.cpp

parent 14cb538f
...@@ -276,7 +276,7 @@ int deform_conv_backward_input_cuda( ...@@ -276,7 +276,7 @@ int deform_conv_backward_input_cuda(
for (int elt = 0; elt < batchSize / im2col_step; elt++) for (int elt = 0; elt < batchSize / im2col_step; elt++)
{ {
columns = columns.addmm_(weight.flatten(1).transpose(0, 1), gradOutputBuffer[elt].flatten(1)); columns = columns.addmm_(weight.flatten(1).transpose(0, 1), gradOutputBuffer[elt].flatten(1), 0.0f, 1.0f);
deformable_col2im_coord( deformable_col2im_coord(
columns, input[elt], offset[elt], columns, input[elt], offset[elt],
......
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