search

fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.
Apache License 2.0
3.14k stars 513 forks source link

Why the sampling_offset bias don't get updates during training #222

Open Levishery opened 8 months ago

Levishery commented 8 months ago

in class MSDeformAttn reset_parameters line 69

        with torch.no_grad():
            self.sampling_offsets.bias = nn.Parameter(grid_init.view(-1))

Why the sampling_offset bias don't get updated during training? I saw that in some other implementations, this parameter would be updated.

Thanks!