Как прочитать контрольную точку Keras в тензорном потоке?

Question

Я использовал ch_callback = ModelCheckpoint('./foo.bar') с моим model.fit().И я получил именно то, что я просил, а именно файл './foo.bar'.

• В каком формате он на самом деле?
• Как можно использовать эту контрольную точку, где я могу ее загрузить?
• Самое главное, могу ли я преобразовать его в собственный формат контрольной точки тензорного потока?

Answer 1

1. Контрольная точка Keras имеет формат .hdf5 или .h5.
2. Вы можете загрузить контрольную точку keras, используя tf.keras.models.load_model("model.h5").
3. Если вы хотите преобразовать контрольную точку Keras в контрольную точку TF, вы можете загрузить свою модель Keras (с бэкэндом Keras), а затем экспортировать контрольную точку TF графика TF, созданного при загрузке модели Keras.

model = keras.models.load_model("model.h5")
sess = keras.backend.get_session()
saver = tf.train.Saver()
save_path = saver.save(sess, "/tmp/model.ckpt")