Keras ModelCheckpoint не сохраняет файлы на Windows - PullRequest
0 голосов
/ 18 января 2020

Я пытался сохранить модель с keras.ModelCheckpoint, но даже если она печатает Epoch 00001: saving model to cp.ckpt et c. он не создает никаких файлов. Есть идеи?

cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath='cp.ckpt', save_weights_only=True, verbose=1)
model_history = model.fit(train_data, epochs=10, steps_per_epoch=200, validation_data=val_data, validation_steps=50, callbacks=[cp_callback])

1 Ответ

1 голос
/ 18 января 2020

Убедитесь, что каталог, в котором вы ищете файл, совпадает с тем, откуда запускается скрипт. Это может быть обманчиво, если вы вызываете его из модуля или другими косвенными способами.

Вы можете попробовать следующее:

  1. print(os.getcwd()), чтобы проверить фактическое каталог, в котором хранится файл

  2. Используйте полный набор патчей на входе filepath, чтобы убедиться, что это каталог, который вы ожидаете.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...