diff --git a/paddlers/custom_models/cd/changestar.py b/paddlers/custom_models/cd/changestar.py index b9950a1..22be256 100644 --- a/paddlers/custom_models/cd/changestar.py +++ b/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(