Modify seresnet after merging #9

2019-03-27 14:34:12 +08:00 · 2019-03-27 14:34:12 +08:00 · 698f85c621
parent 2eb362589b
commit 698f85c621
1 changed files with 1 additions and 1 deletions
--- a/modeling/baseline.py
+++ b/modeling/baseline.py
@ -86,7 +86,7 @@ class Baseline(nn.Module):
                              last_stride=last_stride)
        elif model_name == 'se_resnet152':
            self.base = SENet(block=SEResNetBottleneck, 
-                              layers=[3, 4, 36, 3], 
+                              layers=[3, 8, 36, 3],
                              groups=1, 
                              reduction=16,
                              dropout_p=None,