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):
super(CAFPN, self).__init__()
self.out_channels = out_channels
self.ins_conv = []
self.inp_conv = []
self.ins_conv = nn.LayerList()
self.inp_conv = nn.LayerList()
for i in range(len(in_channels)):
self.ins_conv.append(