Skip to content

Commit be26c0a

Browse files
authored
Add todo
#1
1 parent 0f51e74 commit be26c0a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ def __init__(self, scale=1.0, in_channels=3, c_tag=0.5, num_classes=1000, activa
182182
self.num_of_channels = {0.5: [24, 48, 96, 192, 1024], 1: [24, 116, 232, 464, 1024],
183183
1.5: [24, 176, 352, 704, 1024], 2: [24, 244, 488, 976, 2048]}
184184
self.c = [_make_divisible(chan, groups) for chan in self.num_of_channels[scale]]
185-
self.n = [3, 8, 3]
185+
self.n = [3, 8, 3] # TODO: should be [3,7,3]
186186
self.conv1 = nn.Conv2d(in_channels, self.c[0], kernel_size=3, bias=False, stride=2, padding=1)
187187
self.bn1 = nn.BatchNorm2d(self.c[0])
188188
self.maxpool = nn.MaxPool2d(kernel_size=3, stride=2)

0 commit comments

Comments
 (0)