Update modeling_diffusiondet.py

modeling_diffusiondet.py

@@ -128,7 +128,7 @@ class DiffusionDet(PreTrainedModel):
 
     def _init_weights(self, module):
         if isinstance(module, nn.Linear) or isinstance(module, nn.Conv2d):
-            torch.nn.init.kaiming_normal_(m.weight, mode='fan_in', nonlinearity='relu')
+            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)
         elif isinstance(module, nn.BatchNorm2d):