Проблема с использованием keras.callbacks.ModelCheckpoint с Python 3.5 - PullRequest
0 голосов
/ 17 февраля 2020

Я сохраняю модель в конце каждой тренировочной эпохи, как описано ниже.

Я использую две Windows 10 машины с Python 3.5.6 и 3.6.3 соответственно.

На машине с Python 3.6.3 я могу запустить это без проблем.

На машине с Python 3.5.6, после завершения первой эпохи обучения, я получаю пустой файл «checkpoint-01.hdf5» и консольное сообщение «Процесс завершен с кодом выхода -1073740940»

Может кто-нибудь объяснить это?

from keras.callbacks import ModelCheckpoint

checkpoint = ModelCheckpoint(filepath="./models/checkpoint-{epoch:02d}.hdf5",
                                 monitor='val_loss',
                                 verbose=0,
                                 save_best_only=False,
                                 save_weights_only=False,
                                 mode='auto',
                                 period=1)

model.fit(data.train_data,
              data.train_labels,
              batch_size=batch_size,
              validation_data=(data.validation_data, data.validation_labels),
              nb_epoch=50,
              shuffle=True,
              callbacks=[checkpoint])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...