Update models/GroundingDINO/ms_deform_attn.py

models/GroundingDINO/ms_deform_attn.py

@@ -51,6 +51,7 @@ class MultiScaleDeformableAttnFunction(Function):
         im2col_step,
     ):
         ctx.im2col_step = im2col_step
+        print("getting forward output")
         output = _C.ms_deform_attn_forward(
             value,
             value_spatial_shapes,
@@ -59,6 +60,7 @@ class MultiScaleDeformableAttnFunction(Function):
             attention_weights,
             ctx.im2col_step,
         )
+        print("got msdeform forward output")
         ctx.save_for_backward(
             value,
             value_spatial_shapes,