resnet=torchvision.models.resnet152(pretrained=True) # 原本為1000類,改為10類 resnet.fc=torch.nn.Linear(2048,10) 其中使用了pretrained參數(shù),會直接加載預(yù)訓(xùn)練模型,內(nèi)部實現(xiàn)和前文提到的加載預(yù)訓(xùn)練的方法一樣。因為是先加載的預(yù)訓(xùn)練參數(shù),相當(dāng)于模型中已經(jīng)有參數(shù)了,所以替換掉最后一層即可。OK! 以上就是本文...
www.dbjr.com.cn/article/1702...htm 2025-5-8