Saltar al contenido
Proyecto IA

← Volver al diario

Hecho

Cuarto ensayo, ya en la nube: 2,4 veces más rápido y la mejor pérdida hasta ahora (perplejidad 21,6)

Por decisión del fundador, el entrenamiento del modelo propio ha pasado del portátil a la GPU de Scaleway. El cuarto ensayo es una segunda época sobre los pesos del tercero: mismo modelo de 112 millones de parámetros, mismos 207 millones de tokens en otro orden, 6.000 pasos más con la mitad de tasa de aprendizaje.

Curva de pérdida del ensayo v4 (segunda época): entrenamiento en azul, validación en naranja
Curva de pérdida del ensayo v4 (segunda época): entrenamiento en azul, validación en naranja
Portátil (RTX 5050, 8 GB)Nube (L4, 24 GB)
Segundos por paso de 32.768 tokens3,421,36
Lote por GPU2 secuencias × 16 acumulaciones8 secuencias × 4 acumulaciones
Duración de 6.000 pasos~5,8 h2,27 h
Costecéntimos de luz~1,80 €

La primera evaluación salió idéntica en las dos máquinas (3,299 en el paso 1.000), que es la comprobación de que el cálculo es el mismo y solo cambia la velocidad.

Resultado

Pasov3 (primera época)v4 (segunda época)
1.0004,173,30
3.0003,493,18
6.0003,243,07

Perplejidad de 25 a 21,6. La segunda pasada por los mismos datos todavía mejora, pero cada vez menos: la curva se está aplanando. Es la señal de que con este corpus el modelo pequeño está cerca de su techo y de que lo siguiente que hay que escalar son los datos, no las épocas.

Mientras entrenaba

El motor de inferencia y el reordenador estuvieron parados para dejar la GPU libre (el resto del servicio siguió en marcha). Al terminar se volvieron a levantar, y la consulta pública de normativa respondió desde la web recién publicada. Esta noche el servidor se apaga hasta mañana.

modelopreentrenamientoinfraestructuramedidas