Update modeling_diffusiondet.py

modeling_diffusiondet.py

@@ -129,8 +129,8 @@ class DiffusionDet(PreTrainedModel):
     def _init_weights(self, module):
         if isinstance(module, nn.Linear) or isinstance(module, nn.Conv2d):
             torch.nn.init.kaiming_normal_(module.weight, mode='fan_in', nonlinearity='relu')
-        if module.bias is not None:
-            torch.nn.init.constant_(module.bias, 0)
+            if module.bias is not None:
+                torch.nn.init.constant_(module.bias, 0)
         elif isinstance(module, nn.BatchNorm2d):
             torch.nn.init.constant_(module.weight, 1)
             torch.nn.init.constant_(module.bias, 0)