torch在module中更正fc或者conv层
对于torch每层:
当设置第7层为1000时候,下一层为读入的为1000,所以fine-tune时候,需要重置2层。
vgg.7.weight
torch.Size([1000, 128, 3, 3])
vgg.7.bias
torch.Size([1000])
vgg.10.weight
torch.Size([256, 1000, 3, 3])
vgg.10.bias
torch.Size([256])
torch在module中更正fc或者conv层
对于torch每层:
当设置第7层为1000时候,下一层为读入的为1000,所以fine-tune时候,需要重置2层。
vgg.7.weight
torch.Size([1000, 128, 3, 3])
vgg.7.bias
torch.Size([1000])
vgg.10.weight
torch.Size([256, 1000, 3, 3])
vgg.10.bias
torch.Size([256])