Bläddra i källkod

[Fix] Fix changestar init

Bobholamovic 3 år sedan
förälder
incheckning
ca1051e5eb
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      paddlers/custom_models/cd/changestar.py

+ 2 - 2
paddlers/custom_models/cd/changestar.py

@@ -28,7 +28,7 @@ class _ChangeStarBase(nn.Layer):
 
     def __init__(self, seg_model, num_classes, mid_channels, inner_channels,
                  num_convs, scale_factor):
-        super(_ChangeStarBase, self).__init__(_ChangeStarBase, self)
+        super(_ChangeStarBase, self).__init__()
 
         self.extract = seg_model
         self.detect = ChangeMixin(
@@ -63,7 +63,7 @@ class _ChangeStarBase(nn.Layer):
 
 class ChangeMixin(nn.Layer):
     def __init__(self, in_ch, out_ch, mid_ch, num_convs, scale_factor):
-        super(ChangeMixin, self).__init__(ChangeMixin, self)
+        super(ChangeMixin, self).__init__()
         convs = [Conv3x3(in_ch, mid_ch, norm=True, act=True)]
         convs += [
             Conv3x3(