Sauvegarde un instantanné du modèle

🌐

Checkpoints model weights

Sauvegarde l'instantanné du modèle selon la métrique spécifiée et le comportement.

Utilisation

luz_callback_model_checkpoint(
  path,
  monitor = "valid_loss",
  save_best_only = FALSE,
  mode = "min",
  min_delta = 0
)

Arguments

path

Chemin pour sauvegarder le modèle sur disque dur. Le chemin est interpolé avec glue, donc vous pouvez utiliser tout attribut dans le ctx en utilisant '{ctx$epoch}'. Par exemple, les valeurs epoch et monitor sont déjà présentes dans l'environnement. Si le chemin spécifié est un répertoire (se termine par / ou \), alors les modèles sont sauvegardés avec le nom fourni par epoch-{epoch:02d}-{self$monitor}-{monitor:.3f}.pt. Voir les exemples. Vous pouvez utiliser sprintf() pour formater rapidement les valeurs, par exemple:'{epoch:02d}'.

monitor

Une chaîne au format <set>_<metric> où <set> peut être 'train' ou 'valid' et <metric> est l'abréviation de toute métrique suivie pendant l'entraînement. Le nom de la métrique est insensible à la casse.

save_best_only

Si TRUE, les modèles ne sont sauvegardés que si ils améliorent un modèle précédemment enregistré.

mode

Spécifie la direction considérée comme une amélioration. Par défaut, 'min' est utilisé. Cela peut également être 'max' (lorque on cherche à maximiser la métrique) et 'zero' (plus proche de zéro est mieux).

min_delta

Différence minimale à considérer comme une amélioration. Seulement utilisée lorsque save_best_only=TRUE.

Note

mode et min_delta ne sont utilisés que lorsque save_best_only=TRUE. save_best_only écrasera les modèles enregistrés si le paramètre path ne différencie pas les noms des modèles par epochs.

Voir la vignette sur les instantanés pour plus de détails.

Voir également

Autres callbacks luz: luz_callback_auto_resume(), luz_callback_csv_logger(), luz_callback_early_stopping(), luz_callback_interrupt(), luz_callback_keep_best_model(), luz_callback_lr_scheduler(), luz_callback_metrics(), luz_callback_mixup(), luz_callback_model_checkpoint(), luz_callback_profile(), luz_callback_progress(), luz_callback_resume_from_checkpoint(), luz_callback_train_valid(), luz_callback()

Exemples

luz_callback_model_checkpoint(path= "path/to/dir")
#> Error in luz_callback_model_checkpoint(path = "path/to/dir"): impossible de trouver la fonction "luz_callback_model_checkpoint"
luz_callback_model_checkpoint(path= "path/to/dir/epoch-{epoch:02d}/model.pt")
#> Error in luz_callback_model_checkpoint(path = "path/to/dir/epoch-{epoch:02d}/model.pt"): impossible de trouver la fonction "luz_callback_model_checkpoint"
luz_callback_model_checkpoint(path= "path/to/dir/epoch-{epoch:02d}/model-{monitor:.2f}.pt")
#> Error in luz_callback_model_checkpoint(path = "path/to/dir/epoch-{epoch:02d}/model-{monitor:.2f}.pt"): impossible de trouver la fonction "luz_callback_model_checkpoint"