CommencezCommencez gratuitement

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

Voir le cours

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"
}
Modifier et exécuter le code