Začněte nyníZačněte zdarma

Ukládání vlastních receptur

Zákazník tě nyní požádal o úpravu požadavků. Tentokrát chce zvýšit počet parametrů a použít model Llama 3.2 s 3B parametry. Proveď tuto změnu ve svém slovníku a ulož ho jako soubor YAML.

Knihovna yaml je již naimportována.

Toto cvičení je součástí kurzu

Fine-Tuning s Llama 3

Zobrazit kurz

Pokyny k cvičení

  • Do svého slovníku zadej nový požadavek na model — torchtune.models.llama3_2.llama3_2_3b.
  • Ulož požadavky jako soubor YAML s názvem custom_recipe.yaml.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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:
    ____
Upravit a spustit kód