Définir des recettes personnalisées
Vous effectuez l'ajustement fin d'un modèle Llama préentraîné pour un client qui exige des paramètres précis. Vous prévoyez d'utiliser TorchTune pour l'ajustement fin et devez donc préparer un dictionnaire Python afin d'y consigner les exigences de la recette personnalisée que vous utiliserez pour lancer la tâche d'ajustement fin.
Cette activité fait partie du cours
Ajustement fin avec Llama 3
Instructions de l’exercice
- Indiquez les exigences du client dans votre dictionnaire : d'abord, ajoutez le modèle
torchtune.models.llama3_2.llama3_2_1b. - Ajoutez une taille de lot de 8 et un périphérique GPU.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
config_dict = {
# Define the model
____,
# Define the batch size
____,
# Define the device type
____,
"epochs": 15,
"optimizer": {"_component_": "bitsandbytes.optim.PagedAdamW8bit", "lr": 3e-05},
"dataset": {"_component_": "custom_dataset"},
"output_dir": "/tmp/finetune_results"
}