用于记录:
pretrained_model = models.vgg16(True).features[:-1]
print(pretrained_model)
就是把vgg16最后一层不要了
pretrained_model = models.vgg16
用于记录:
pretrained_model = models.vgg16(True).features[:-1]
print(pretrained_model)
就是把vgg16最后一层不要了
pretrained_model = models.vgg16