我正在關注this blog,並且因爲它在鏈接的博客中使用而難以實施保存檢查點。在第23行使用:Keras保存檢查點
filepath="weights-improvement-{epoch:02d}-{val_acc:.2f}.hdf5"
。
於是,我就調整代碼一點點更有活力:
filepath = '{0}/checkpoints/checkpoint-{epoch:02d}-{val_loss:.2f}.hdf5'.format(directory)
。
,我想給定架構的所有檢查站存放在1個目錄,例如:./architecture1/checkpoints/
,但我得到了以下錯誤:KeyError: 'epoch'
。我在這裏做錯了什麼?
P.S. .: filepath = "./checkpoints/checkpoint-{epoch:02d}-{val_loss:.2f}.hdf5"
的作品,但它保存了我不想要的1個目錄中的所有檢查點。
你嘗試導入操作系統; filepath = os.path.join(目錄,'checkpoints','checkpoint- {epoch:02d} - {val_loss:.2f} .hdf5')? –
如何將一個目錄分配給模型? –
其實我只是發現這個問題可以用簡單的字符串連接來解決。 – narn