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')
当我尝试加载最佳权重时,显示没有该文件或目录。我应该如何修改?非常感谢大家!
回答:
你在文件名中有一个拼写错误;你保存的是
filepath="best_weight.hdf5"
而你尝试加载的是 'best_weights.hdf5'
。
从文件名中删除最后的 s
,即:
classifier.load_weights('best_weight.hdf5')