Como posso usar tf.keras.retornos de chamada.ModelCheckpoint em Keras Sintonizador?

0

Pergunta

Então, eu quero usar tf.keras.retornos de chamada.ModelCheckpoint em Keras Sintonizador, mas A maneira como você escolha o caminho onde salvar os pontos de controle, não permitem que você salvá-lo como um arquivo com um certo nome, um nome associado para o julgamento e execução de ponto de verificação, apenas associado a uma época.

Isto é, se eu simplesmente colocar esse retorno no Keras Sintonizador, no momento em que os pontos de verificação guardar acontece, no final, eu não sei como associar os pontos de verificação guardado para um ensaio e avaliação de execução, apenas para época.

1

Melhor resposta

0

Você pode usar tf.keras.callbacks.ModelCheckpoint para Keras tuner da mesma forma como é usado em outros modelos para salvar postos de controle.

Após o treinamento o modelo com hiperparâmetros obtidos a partir da pesquisa conforme este modelo, você pode definir o modelo de pontos de verificação e salve-o como abaixo:

hypermodel = tuner.hypermodel.build(best_hps)

# Retrain the model
hypermodel.fit(img_train, label_train, epochs=best_epoch, validation_split=0.2)

import os
checkpoint_path = "training_1/cp.ckpt"
checkpoint_dir = os.path.dirname(checkpoint_path)

# Create a callback that saves the model's weights
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath=checkpoint_path,
                                                 save_weights_only=True,
                                                 verbose=1)
history = hypermodel.fit(img_train, label_train, epochs=5, validation_split=0.2, callbacks=[cp_callback])
os.listdir(checkpoint_dir)

# Re-evaluate the model
loss, acc = hypermodel.evaluate(img_test, label_test, verbose=2)
print("Restored model, accuracy: {:5.2f}%".format(100 * acc))

# Loads the weights
hypermodel.load_weights(checkpoint_path)

# Re-evaluate the model
loss, acc = hypermodel.evaluate(img_test, label_test, verbose=2)
print("Restored model, accuracy: {:5.2f}%".format(100 * acc))

Por favor consulte este link para mais inofrmation em salvar e carregar o modelo de pontos de verificação.

2021-12-06 16:04:19

Em outros idiomas

Esta página está em outros idiomas

Русский
..................................................................................................................
Italiano
..................................................................................................................
Polski
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................