分类：fine-tune

2021
08-06

pytorch fine-tune 预训练的模型操作

之一:torchvision中包含了很多预训练好的模型，这样就使得fine-tune非常容易。本文主要介绍如何fine-tunetorchvision中预训练好的模型。安装pipinstalltorchvision如何fine-tune以resnet18为例：fromtorchvisionimportmodelsfromtorchimportnnfromtorchimportoptimresnet_model=models.resnet18(pretrained=True)#pretrained设置为True，会自动下载模型所对应权重，并加载到模型中#也可以自己下载权... 继续阅读 >