fix export

This commit is contained in:
LDOUBLEV 2022-04-26 19:49:08 +08:00
parent 10c5773b9b
commit 293e85bc6d

View File

@ -144,8 +144,8 @@ class CAFPN(nn.Layer):
def __init__(self, in_channels, out_channels, shortcut=True, **kwargs): def __init__(self, in_channels, out_channels, shortcut=True, **kwargs):
super(CAFPN, self).__init__() super(CAFPN, self).__init__()
self.out_channels = out_channels self.out_channels = out_channels
self.ins_conv = [] self.ins_conv = nn.LayerList()
self.inp_conv = [] self.inp_conv = nn.LayerList()
for i in range(len(in_channels)): for i in range(len(in_channels)):
self.ins_conv.append( self.ins_conv.append(