Keras:我使用ModelCheckpoint保存模型,但是在目录中找不到模型文件

时间:2019-10-21 03:49:28

标签: python keras

我使用ModelCheckpoint保存模型,但是在目录中找不到任何模型文件。

我搜索了很多博客,但是它们没用

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=0, 
save_best_only=False, mode='max', period=1)
callbacks_list = [checkpoint]

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

model.fit(train_images, train_labels, epochs=3)

我想找到模型文件。

1 个答案:

答案 0 :(得分:0)

您忘了在模型中添加回调:

model.fit(train_images, train_labels, epochs=3, callbacks=callbacks_list)