Retour aux cas pratiques
PyTorchAvancé

Correction d’un modèle avec learning rate inadapté

Scénario

Contexte : L’entraînement d’un modèle PyTorch est instable et la perte oscille fortement. Le Problème : Le learning rate choisi est trop élevé. Extrait logique actuelle : optimizer = torch.optim.Adam(model.parameters(), lr=0.1) Contraintes : Tester plusieurs valeurs de learning rate. Introduire scheduler si nécessaire. Comparer stabilité des pertes. Livrable attendu : Proposer une stratégie d’optimisation du learning rate améliorant convergence et performance.

Compétences ciblées

learning-rateoptimizertrainingbugfixoptimization
    Correction d’un modèle avec learning rate inadapté — mise en situation PyTorch | CodeDuel