diff --git a/ssd.py b/ssd.py index cd570064c..ef92e91a0 100644 --- a/ssd.py +++ b/ssd.py @@ -163,7 +163,7 @@ def add_extras(cfg, i, batch_norm=False): def multibox(vgg, extra_layers, cfg, num_classes): loc_layers = [] conf_layers = [] - vgg_source = [24, -2] + vgg_source = [21, -2] for k, v in enumerate(vgg_source): loc_layers += [nn.Conv2d(vgg[v].out_channels, cfg[k] * 4, kernel_size=3, padding=1)]