【发布时间】:2017-09-01 09:15:52
【问题描述】:
我刚刚第一次使用ModelCheckpoint 函数来保存最佳模型(best_model = True)并想测试它的性能。保存模型时,它说val_acc 在保存之前为 83.3%。我加载了模型并在validation_generator 上使用了evaluate_generator,但val_acc 的结果是0.639。我感到困惑并再次使用它并得到 0.654,然后是 0.647、0.744 等等。我在我的 PC 上测试了相同的配置(没有 GPU),它始终显示相同的结果(有时可能是小的舍入误差)
- 为什么不同 evaluate_generator 执行之间的结果仅在 GPU 上有所不同?
- 为什么模型 val_acc 与报告的不同?
我正在使用 Keras 的 Tensorflows 实现。
model.compile(loss='categorical_crossentropy',
optimizer=optimizers.SGD(lr=1e-4, momentum=0.9),
metrics=['accuracy'])
checkpointer = ModelCheckpoint(filepath='/tmp/weights.hdf5', monitor = "val_acc", verbose=1, save_best_only=True)
# prepare data augmentation configuration
train_datagen = ImageDataGenerator(
rescale = 1./ 255,
shear_range = 0.2,
zoom_range = 0.2,
horizontal_flip = True)
test_datagen = ImageDataGenerator(rescale=1. / 255)
train_generator = train_datagen.flow_from_directory(
train_data_dir,
target_size = (img_height, img_width),
batch_size = batch_size)
validation_generator = test_datagen.flow_from_directory(
validation_data_dir,
target_size = (img_height, img_width),
batch_size = batch_size)
# fine-tune the model
model.fit_generator(
train_generator,
steps_per_epoch = math.ceil(train_samples/batch_size),
epochs=100,
workers = 120,
validation_data=validation_generator,
validation_steps=math.ceil(val_samples/batch_size),
callbacks=[checkpointer])
model.load_weights(filepath='/tmp/weights.hdf5')
model.predict_generator(validation_generator, steps = math.ceil(val_samples/batch_size) )
temp_model = load_model('/tmp/weights.hdf5')
temp_model.evaluate_generator(validation_generator, steps = math.ceil(val_samples/batch_size), workers = 120)
>>> [2.1996076788221086, 0.17857142857142858]
temp_model.evaluate_generator(validation_generator, steps = math.ceil(val_samples/batch_size), workers = 120)
>>> [2.2661823204585483, 0.25]
【问题讨论】:
-
您的生成器是否每次都返回相同的验证集?它如何选择样本有随机性吗?
-
在代码中,我使用内置的 flow_from_directory。 train 和 test 这两个文件夹是分开的。我认为它使用一切=>没有随机性。 train_data_dir = '/data/datasets/NAME/train' validation_data_dir = '/data/datasets/NAME/test'
-
val_samples是否等于validation_data_dir下的图片文件总数? -
这实际上是我犯的一个错误,但即使在一切正常的 train_generator 中,错误仍然存在。
-
改正后错误依旧
标签: python tensorflow neural-network keras tensorflow-gpu