Zacznij terazZacznij za darmo

Zapisywanie niestandardowych przepisów

Klient zgłosił nowe wymagania. Tym razem chce zwiększyć liczbę parametrów i użyć modelu Llama 3.2 z 3 miliardami parametrów. Wprowadź tę zmianę do swojego słownika, a następnie zapisz go jako plik YAML.

Biblioteka yaml została już zaimportowana.

To ćwiczenie jest częścią kursu

Fine-Tuning z Llama 3

Zobacz kurs

Instrukcje do ćwiczenia

  • Podaj nowe wymaganie dotyczące modelu – torchtune.models.llama3_2.llama3_2_3b – w swoim słowniku.
  • Zapisz wymagania jako plik YAML o nazwie custom_recipe.yaml.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

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:
    ____
Edytuj i uruchom kod