callback tfevent

🌐

tfevents callback

Enregistre des métriques et d'autres informations du modèle sous forme de fichier tfevents. Quand tensorboard est installé, il peut être utilisé pour visualiser les résultats.

Utilisation

luz_callback_tfevents(logdir = "logs", histograms = FALSE, ...)

Arguments

logdir

Un répertoire où sera écrit le journal.

histograms

Un booléen spécifiant si les histogrammes des poids du modèle doivent être enregistrés. Peut aussi être un vecteur de chaîne de caractères spécifiant les noms des paramètres à enregistrer (noms identiques à names(model$parameters)).

...

Inutilisé

Détails

tensorboard --logdir=logs

Exemples

if (torch::torch_is_installed()) {
library(torch)
x <- torch_randn(1000, 10)
y <- torch_randn(1000, 1)

model <- nn_linear %>%
  setup(loss = nnf_mse_loss, optimizer = optim_adam) %>%
  set_hparams(in_features = 10, out_features = 1) %>%
  set_opt_hparams(lr = 1e-4)

tmp <- tempfile()

model %>% fit(list(x, y), valid_data = 0.2, callbacks = list(
  luz_callback_tfevents(tmp, histograms = TRUE)
))
}
#> Error in set_opt_hparams(., lr = 1e-04): impossible de trouver la fonction "set_opt_hparams"