Actualización de peso para vgg16

Cuando obtengamos los parámetros del modelo de red vgg16, si los obtenemos de la siguiente manera, se producirá un error.

vgg16_false = torchvision.models.vgg16(pretrained=False)
vgg16_true = torchvision.models.vgg16(pretrained=True)

 El código se puede cambiar de la siguiente manera:

vgg16_false = torchvision.models.vgg16(weights=None)

vgg16_true = torchvision.models.vgg16(weights=VGG16_Weights.DEFAULT)

Si lo vuelves a ejecutar, no reportarás un error.

Motivo del error: el parámetro preentrenado está en desuso desde la versión 0.13 y se eliminará después de la 0.15. Utilice "pesos" en su lugar.

Supongo que te gusta

Origin blog.csdn.net/weixin_67957872/article/details/127173609
Recomendado
Clasificación