Răsfoiți Sursa

[Fix] add init_weight method of DSIFN

Bobholamovic 3 ani în urmă
părinte
comite
fd4118afe3
1 a modificat fișierele cu 6 adăugiri și 0 ștergeri
  1. 6 0
      paddlers/models/cd/models/dsifn.py

+ 6 - 0
paddlers/models/cd/models/dsifn.py

@@ -88,6 +88,8 @@ class DSIFN(nn.Layer):
         self.bn_sa5 = make_norm(16)
         self.o5_conv4 = Conv1x1(16, num_classes)
 
+        self.init_weight()
+
     def forward(self, t1, t2):
         # Extract bi-temporal features.
         with paddle.no_grad():
@@ -173,6 +175,10 @@ class DSIFN(nn.Layer):
 
         return out5, out4, out3, out2, out1
 
+    def init_weight(self):
+        # Do nothing
+        pass
+
 
 class VGG16FeaturePicker(nn.Layer):
     def __init__(self, indices=(3,8,15,22,29)):