HichTala
/

DiffusionDet

Object Detection

Model card Files Files and versions

HichTala commited on Apr 18, 2025

Commit

d30f52f

·

verified ·

1 Parent(s): 5ac9c4e

Update modeling_diffusiondet.py

Files changed (1) hide show

modeling_diffusiondet.py +1 -1

modeling_diffusiondet.py CHANGED Viewed

@@ -127,7 +127,7 @@ 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 (

         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 (