ComenzarEmpieza gratis

Guardar recetas personalizadas

El cliente te ha pedido ahora una modificación en los requisitos. Esta vez, quiere aumentar el número de parámetros y usar el modelo Llama 3.2 con 3B de parámetros. Aplica esta modificación a tu diccionario y guárdalo como un archivo YAML.

La biblioteca yaml ya se ha importado.

Este ejercicio forma parte del curso

Ajuste fino con Llama 3

Ver curso

Instrucciones del ejercicio

  • Especifica en tu diccionario el nuevo requisito de modelo, torchtune.models.llama3_2.llama3_2_3b.
  • Guarda los requisitos en un archivo YAML llamado custom_recipe.yaml.

Ejercicio interactivo práctico

Prueba este ejercicio y completa el código de muestra.

config_dict = {
    # Update the model
    ____,
    "batch_size": 8,
    "device": "cuda",
    "optimizer": {"_component_": "bitsandbytes.optim.PagedAdamW8bit", "lr": 3e-05},
    "dataset": {"_component_": "custom_dataset"},
    "output_dir": "/tmp/finetune_results"
}

# Save the updated configuration to a new YAML file
with open("custom_recipe.yaml", "w") as yaml_file:
    ____
Editar y ejecutar código