Skip to content

Commit 0e2196b

Browse files
committed
fix error
1 parent ee6d8d0 commit 0e2196b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

timm/models/dpn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,8 @@ def __init__(
8787
self.c1x1_c2 = create_conv2d(num_3x3_b, inc, kernel_size=1)
8888
else:
8989
self.c1x1_c = BnActConv2d(in_chs=num_3x3_b, out_chs=num_1x1_c + inc, kernel_size=1, stride=1)
90-
self.c1x1_c1 = None
91-
self.c1x1_c2 = None
90+
self.c1x1_c1 = nn.Identity()
91+
self.c1x1_c2 = nn.Identity()
9292

9393
def forward(self, x: torch.Tensor) -> torch.Tensor:
9494
x_s = self.c1x1_w(x)

0 commit comments

Comments
 (0)