¿Qué parámetros se deben completar para torchvision.models.vgg16() en pytorch?
En torchvision.models.vgg16() en PyTorch, puede decidir si cargar los pesos previamente entrenados configurando el parámetro preentrenado, que acepta un valor booleano: Verdadero significa cargar los pesos previamente entrenados, Falso significa no cargar. El mensaje de advertencia dice que después de la versión 0.13, "peso" ya no se usa como parámetro, por lo que se deben usar argumentos de palabras clave.
Puedes llamar al modelo VGG16 de esta manera:
importar torchvision.models como modelos
# Cargar el modelo VGG16 usando pesos previamente entrenados
vgg16_pretrained = models.vgg16( pretrained=True)
# Cargar modelo VGG16 sin pesos previamente entrenados
vgg16_not_pretrained = models.vgg16(pretrained=False)
De esta forma podrás evitar avisos y utilizar correctamente el modelo VGG16 con o sin pesas previamente entrenadas.