错误：使用pytorch在CNN（vgg16）中添加图层时，索引7超出范围

Question

我正在使用pytorch实施转移学习。 这是我的初始代码：（仅在最后一层进行更改）

from torchvision import datasets, models, transforms
model_ft = models.vgg16_bn(pretrained=True)
model_ft.classifier[6] = nn.Linear(in_features=4096,
                               out_features=2,
                               bias=True) 

问题是结果不令人满意。 这就是为什么我想添加更多的图层，以便不必从4096跳到2。

from torchvision import datasets, models, transforms
model_ft = models.vgg16_bn(pretrained=True)
model_ft.classifier[6] = nn.Linear(in_features=4096,
                                   out_features=1024,
                                   bias=True)
model_ft.classifier[7] = nn.ReLU()
model_ft.classifier[8] = nn.Dropout(p=0.5)
model_ft.classifier[9] = nn.Linear(in_features=1024,
                                   out_features=10,
                                   bias=True)
Traceback (most recent call last):

  File "<ipython-input-14-519d30b6f8d1>", line 5, in <module>
    model_ft.classifier[7] = nn.ReLU()

  File "C:\Users\frmas\anaconda3\envs\tfgpu\lib\site-packages\torch\nn\modules\container.py", line 
86, in __setitem__
    key = self._get_item_by_idx(self._modules.keys(), idx)

  File "C:\Users\frmas\anaconda3\envs\tfgpu\lib\site-packages\torch\nn\modules\container.py", line 
74, in _get_item_by_idx
    raise IndexError('index {} is out of range'.format(idx))

IndexError: index 7 is out of range

为什么我不能在vgg16分类中添加更多层。 怎么解决呢？