Salvarea rețetelor personalizate
Clientul ți-a solicitat acum o modificare a cerințelor. De data aceasta, dorește creșterea numărului de parametri și utilizarea modelului Llama 3.2 cu 3B parametri. Faci această modificare în dicționarul tău, apoi îl salvezi ca fișier YAML.
Biblioteca yaml a fost deja importată.
Acest exercițiu face parte din cursul
Fine-Tuning cu Llama 3
Instrucțiuni pentru exercițiu
- Specifică noua cerință de model,
torchtune.models.llama3_2.llama3_2_3b, în dicționarul tău. - Salvează cerințele ca fișier YAML cu numele
custom_recipe.yaml.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
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:
____