Konfigurace optimizeru
Teď, když máme tréninkovou logiku, je potřeba určit, jak optimalizovat parametry modelu.
V tomto cvičení dokončíš metodu configure_optimizers v modulu PyTorch Lightning určeném pro klasifikaci obrázků. Tvým cílem je nastavit optimizer, který bude během tréninku aktualizovat parametry modelu. K tomu použiješ optimizer Adam s rychlostí učení 1e-3.
Toto cvičení je součástí kurzu
Škálovatelné modely AI s PyTorch Lightning
Pokyny k cvičení
- Vytvoř optimizer Adam s parametry modelu a nastav rychlost učení na
1e-3.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
import torch
def configure_optimizers(self):
# Create an Adam optimizer for model parameters
optimizer = ____
return optimizer