python - Keras 保存检查点

Question

我正在关注 this blog并且我无法实现保存检查点，因为它在链接的博客中使用。在第 23 行，它使用了:

filepath="weights-improvement-{epoch:02d}-{val_acc:.2f}.hdf5"。

所以我尝试稍微调整一下代码以使其更加动态:

filepath = '{0}/checkpoints/checkpoint-{epoch:02d}-{val_loss:.2f}.hdf5'.format(目录)。

我想将给定架构的所有检查点存储在 1 个目录中，例如:./architecture1/checkpoints/

但我收到以下错误:KeyError: 'epoch'。我在这里做错了什么？

P.S.: filepath = "./checkpoints/checkpoint-{epoch:02d}-{val_loss:.2f}.hdf5" 有效，但它会将所有检查点保存在我不知道的 1 个目录中我不想。

Answer 1

如果你想使用format，正确的方法是像这样转义括号:

filepath = '{0}/checkpoints/checkpoint-{{epoch:02d}}-{{val_loss:.2f}}.hdf5'.format(directory)

因此，如果 directory = 'weights'，filepath 将是 'weights/checkpoints/checkpoint-{epoch:02d}-{val_loss:.2f} .hdf5'.

(注意目录是否包含{})