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
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:
____