IniziaInizia gratis

Salvare ricette personalizzate

Il cliente ti ha chiesto una modifica ai requisiti. Questa volta vuole aumentare il numero di parametri e usare il modello Llama 3.2 con 3B di parametri. Apporta questa modifica al tuo dizionario e poi salvalo come file YAML.

La libreria yaml è già stata importata.

Questo esercizio fa parte del corso

Fine-tuning con Llama 3

Visualizza il corso

Istruzioni dell'esercizio

  • Specifica nel tuo dizionario il nuovo requisito del modello, torchtune.models.llama3_2.llama3_2_3b.
  • Salva i requisiti in un file YAML chiamato custom_recipe.yaml.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

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:
    ____
Modifica ed esegui il codice