revise format

This commit is contained in:
rujiao.lrj
2022-11-17 15:48:40 +08:00
parent 3d781dfcde
commit 1cb215422d

View File

@@ -636,7 +636,7 @@ class DLAUp(nn.Module):
IDAUp(3, channels[j], in_channels[j:], scales[j:] // scales[j]),
)
scales[j + 1:] = scales[j]
in_channels[j + 1 :] = [channels[j] for _ in channels[j + 1 :]]
in_channels[j + 1:] = [channels[j] for _ in channels[j + 1:]]
def forward(self, layers):
layers = list(layers)