¿Cómo utilizar pesas previamente entrenadas después de cambiar la red mobilenetv2?
¿El interlocutor quiere preguntar "¿Cómo utilizar pesos previamente entrenados después de cambiar la red mobilenetv2"? 1. Primero, para el modelo de red ajustado, puede establecer el parámetro estricto = Falso en la función Loaded_state_dict ().
2. En segundo lugar, cuando estricto = Verdadero, los parámetros en el archivo de peso cargado deben coincidir completamente con los parámetros del modelo. Después de establecer estricto = Falso, puede usar los datos en el archivo de peso dentro. la capa que coincide con el modelo, el resto se inicializa de forma predeterminada.
3. Finalmente, cuando los pesos previos al entrenamiento no se pueden cargar después de cambiar el modelo de acuerdo con el método anterior, puede considerar obtener los pesos del modelo original y cargarlos en el nuevo modelo.