J'ai construit et entraîné un modèle RNN dans Colaboratory, et je l'ai enregistré à l'aide du rappel de point de contrôle :

tf.keras.callbacks.ModelCheckpoint( filepath=chkpt_p, save_weights_only=False)

Et les données enregistrées ressemblent à ceci :

enter image description here

Si je comprends bien, ce répertoire comprend à la fois le modèle et les poids. Quelle fonction dois-je utiliser pour le restaurer (dans un autre runtime) ?

0
Anna 15 nov. 2020 à 16:15

1 réponse

Meilleure réponse

Cela a fonctionné pour moi:

model = tf.keras.models.load_model(save_dir,compile=False)

J'ai eu un problème avec l'utilisation de tf.saved_model.load, qui ne produit pas d'objet keras, et un TF ne trouvant pas la fonction de perte avec compile=True.

0
Anna 15 nov. 2020 à 14:12