Entrenamiento

Punto de Control del Modelo

Un punto de control del modelo es una captura guardada de los pesos de una red neuronal y el estado del optimizador en un punto específico durante el entrenamiento, permitiendo la reanudación después de fallas de hardware y la selección de la versión de mejor rendimiento a través de pasos de entrenamiento.

Un punto de control del modelo es un archivo o conjunto de archivos serializado que captura el estado completo de una red neuronal en un paso de entrenamiento dado o época: los pesos del modelo, el estado del optimizador (incluyendo buffers de momentum y estadísticas de tasa de aprendizaje adaptativa), la posición del programa de tasa de aprendizaje actual, y el índice de paso o época. Guardar puntos de control a intervalos regulares es práctica estándar en cualquier ejecución de entrenamiento de aprendizaje profundo no trivial.

Los puntos de control se escriben usando formatos de serialización específicos del framework. PyTorch usa archivos .pt o .pth producidos por torch.save; TensorFlow usa directorios SavedModel o fragmentos .ckpt; el ecosistema de Hugging Face ha adoptado ampliamente el formato safetensors, que se carga más rápido y evita las vulnerabilidades de seguridad de la serialización basada en pickle. Para modelos grandes, los archivos de punto de control varían desde unos pocos megabytes para clasificadores pequeños hasta cientos de gigabytes para modelos de lenguaje de 70B+ parámetros, a menudo almacenados como archivos fragmentados en almacenamiento de objetos distribuido.

Los puntos de control sirven múltiples propósitos en la práctica: tolerancia a fallos (reanudar una ejecución de entrenamiento de múltiples semanas después de un fallo de nodo sin perder todo el progreso), selección de modelo (retener el punto de control del paso con la mejor métrica de validación en lugar de tomar ciegamente los pesos finales), y despliegue (usar pesos guardados directamente como el artefacto de inferencia de producción). Promediado de puntos de control—calcular la media elemento a elemento de pesos a través de varios puntos de control recientes—es una técnica adicional a veces usada para mejorar la generalización.

A partir de 2026, Hugging Face Hub aloja millones de puntos de control de modelo públicos, haciendo que el compartir puntos de control sea el estándar de facto para distribución de modelos y reproducibilidad. Plataformas de entrenamiento en la nube como AWS SageMaker, Google Vertex AI, y Azure ML ofrecen gestión integrada de puntos de control con guardados automáticos periódicos a almacenamiento de objetos como S3 o GCS. Para modelos muy grandes, el tiempo requerido para escribir y recargar un punto de control puede en sí convertirse en un cuello de botella, impulsando trabajo en guardas asincrónicas y guardas de delta incremental.

Ejemplo

Durante una ejecución de una semana de fine-tuning de un modelo de lenguaje de 13B parámetros, se guardan puntos de control cada 500 pasos de entrenamiento al almacenamiento en la nube; cuando un nodo GPU falla en el paso 3,200, la ejecución se reanuda desde el punto de control del paso-3,000 en lugar de reiniciar desde cero.

Términos relacionados

← Glosario