Upload modeling_diffusiondet.py

modeling_diffusiondet.py

@@ -126,6 +126,9 @@ class DiffusionDet(PreTrainedModel):
 
         self.criterion = CriterionDynamicK(config, num_classes=self.num_classes, weight_dict=weight_dict)
 
+    def _init_weights(self, module):
+        torch.nn.init.xavier_uniform(module.weight)
+
     def predict_noise_from_start(self, x_t, t, x0):
         return (
                 (extract(self.sqrt_recip_alphas_cumprod, t, x_t.shape) * x_t - x0) /