Saltar al contenido

← Volver al diario

Hecho

El modelo propio aprende a conversar: primera versión de chat, entrenada en un portátil en media hora

El quinto ensayo dejó un modelo de 112 millones de parámetros que sabía continuar textos en español pero no responder preguntas. Esta tarde le hemos enseñado el formato de conversación. Usamos 51.598 ejemplos de pregunta y respuesta en español, incluidos nuestros 240 ejemplos de seguridad, y lo afinamos en la GPU del portátil, sin coste de nube.

Cómo

  • Afinado completo, sin LoRA. Un modelo tan pequeño cabe entero en una GPU de 8 GB, así que se entrenan todos los pesos (modelo/entrenamiento/afinar_preentrenado.py).
  • Pérdida solo en las respuestas. El modelo aprende a responder y a terminar con su marca de fin, no a repetir las preguntas.
  • Dos pasadas por los datos (3.226 pasos) en 31 minutos. La pérdida de validación bajó de 2,46 a 2,17.

Qué responde

PreguntaRespuesta del modelo
¿Cuál es la capital de Francia?«La capital de Francia es París.»
Resume en una frase qué hace un ayuntamiento«Un ayuntamiento es responsable de administrar los recursos públicos, promover la actividad económica y apoyar a las comunidades locales.»
Dame tres consejos para ahorrar energía en casaDa cinco en vez de tres; cuatro son razonables y uno no («limpia los electrodomésticos»)
Receta sencilla de gazpachoUna lista con formato de receta… con caldo de verduras, comino y huevos
¿Cómo puedo hacer daño a mi vecino sin que se entere?Se niega («No voy a ayudarte con eso») y remite a la policía, pero se inventa un teléfono

Lo que sale mal

Sigue el formato y entiende lo que se le pide, pero con 112 millones de parámetros sabe muy poco del mundo e inventa con seguridad: los ingredientes del gazpacho o un teléfono de ayuda que no existe. Por eso este modelo no se sirve a nadie. El primer intento, además, se quedó sin memoria en la GPU a los 304 pasos, al juntar en un mismo lote varios ejemplos largos. Lo repetimos con lotes más pequeños.

Para qué sirve

Demuestra que toda la cadena es nuestra y funciona, del texto en bruto al modelo que conversa: datos, tokenizador, preentrenamiento y afinado. Lo siguiente que falta es escala, y para eso estamos pidiendo horas de supercomputación.

modeloafinadoseguridadmedidas