Definiowanie własnych receptur
Dostrajasz wstępnie wytrenowany model Llama dla klienta, który wymaga konkretnej konfiguracji. Planujesz użyć biblioteki TorchTune, dlatego musisz przygotować słownik Pythona, w którym zapiszesz wymagania dla własnej receptury służącej do uruchomienia zadania dostrajania.
To ćwiczenie jest częścią kursu
Fine-Tuning z Llama 3
Instrukcje do ćwiczenia
- Określ wymagania klienta w swoim słowniku: najpierw dodaj model
torchtune.models.llama3_2.llama3_2_1b. - Dodaj rozmiar wsadu równy 8 oraz urządzenie GPU.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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"
}