Pytorch下微调网络模型进行图像分类

利用ImageNet下的预训练权重采用迁移学习策略，能够实现模型快速训练，提高图像分类性能。下面以vgg和resnet网络模型为例，微调最后的分类层进行分类。 注意，微调只对分类层(也就是全连接层 ...