CommencerCommencer gratuitement

Enregistrement des mesures d'évaluation

Le cursus des mesures de performance vous permet de surveiller les dégradations, et vous pouvez prendre des décisions sur le moment de mettre à jour votre modèle pour maintenir un niveau élevé de précision. Vous décidez de consigner les mesures après que votre modèle a terminé une boucle d'évaluation.

Certaines données ont été préchargées :

  • accelerator est une instance de Accelerator
  • eval_metric est un dictionnaire de mesures telles que accuracy et f1
  • num_epochs est le nombre d'époques

Cet exercice fait partie du cours

Entraînement efficace de modèles d'IA avec PyTorch

Afficher le cours

Instructions

  • Appelez une méthode pour enregistrer les mesures d'évaluation du modèle.
  • Le journal "accuracy" et le score "f1" sont des mesures d'évaluation.
  • Cursus du numéro d'époque à l'aide de epoch de la boucle d'apprentissage.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

accelerator = Accelerator(project_dir=".", log_with="all")
accelerator.init_trackers("my_project")

for epoch in range(num_epochs):
    # Training loop is here
    # Evaluation loop is here
    # Call a method to log metrics
    ____.____({
        # Log accuracy and F1 score as metrics
        "accuracy": ____["accuracy"],
        "f1": ____["f1"],
    # Track the epoch number
    }, ____=____)

accelerator.end_training()
Modifier et exécuter le code