Eigene Rezepte speichern
Der Kunde hat nun eine Anpassung der Anforderungen gewünscht. Diesmal soll die Parameterzahl erhöht und das Llama-3.2-Modell mit 3B Parametern verwendet werden. Du nimmst diese Änderung in deinem Dictionary vor und speicherst es anschließend als YAML-Datei.
Die Bibliothek yaml wurde bereits importiert.
Diese Übung ist Teil des Kurses
Feinabstimmung mit Llama 3
Anleitung zur Übung
- Gib die neue Modellanforderung, das Modell
torchtune.models.llama3_2.llama3_2_3b, in deinem Dictionary an. - Speichere die Anforderungen als YAML-Datei mit dem Namen
custom_recipe.yaml.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
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:
____