resnet152 = models.resnet152 (pretrained = True)
pretrained_dict = resnet152.state_dict ()
"" "Después de cargar el modelo pre-entrenado y los parámetros en torchvision, extraiga los parámetros a través del método state_dict ().
También puede descargar directamente desde el model_zoo:
pretrained_dict = model_zoo.load_url oficial. (model_urls ['resnet152']) "" "
model_dict = model.state_dict ()
Eliminar claves en pretrained_dict que no son parte de model_dict
pretrained_dict = {k: v para k, v en pretrained_dict.items () si k en model_dict}
Actualizar model_dict existente
model_dict.update (pretrained_dict)
Cargue el estado_dict que realmente necesitamos
model.load_state_dict (model_dict)
————————————————
Aviso de copyright: Este artículo es un artículo original del blogger de CSDN "lscelory", siguiendo el acuerdo de copyright CC 4.0 BY-SA Enlace fuente original y esta declaración.
Enlace original: https://blog.csdn.net/lscelory/article/details/81482586
