Keras не может использовать обратный вызов для хранения контрольной точки

Question

train_data = ImageDataGenerator(rescale=1./255)
test_data = ImageDataGenerator(rescale=1./255)
train_set = train_data.flow_from_directory('train_set',
                                           target_size=(28,28),
                                           batch_size=16,
                                           class_mode='categorical')
test_set = train_data.flow_from_directory('test_set',
                                           target_size=(28,28),
                                           batch_size = 16,
                                           class_mode='categorical')
checkpointer = ModelCheckpoint(filepath="best_weight.hdf5",
                               monitor='val_acc',
                               verbose=1,
                               save_best_only=True)
history = classifier.fit_generator(train_set,
                                   steps_per_epoch=210,
                                   epochs=5,
                                   callbacks=[checkpointer],
                                   validation_data=test_set,
                                   validation_steps=90)

classifier.load_weights('best_weights.hdf5')

Когда я пытался загрузить лучшие веса, он не показывает такой файл или каталог.Как мне внести изменения?Большое спасибо, ребята!

Answer 1

У вас есть опечатка в имени файла;вы сохраняете в

filepath="best_weight.hdf5"

при попытке загрузить 'best_weights.hdf5'.

Удалите окончательный s из имени файла, то есть:

classifier.load_weights('best_weight.hdf5')