Skip to content

Commit e670a37

Browse files
committed
fix linter msg
1 parent a1ec1d3 commit e670a37

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

model_constructor/convmixer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,8 @@ def __init__(self, dim: int, depth: int,
9696
stem,
9797
*[nn.Sequential(
9898
Residual(
99-
ConvLayer(dim, dim, kernel_size, act_fn=act_fn, groups=dim, padding="same", bn_1st=bn_1st, pre_act=pre_act)),
99+
ConvLayer(dim, dim, kernel_size, act_fn=act_fn,
100+
groups=dim, padding="same", bn_1st=bn_1st, pre_act=pre_act)),
100101
ConvLayer(dim, dim, kernel_size=1, act_fn=act_fn, bn_1st=bn_1st, pre_act=pre_act))
101102
for i in range(depth)],
102103
nn.AdaptiveAvgPool2d((1, 1)),

0 commit comments

Comments
 (0)