【发布时间】:2018-05-28 19:28:31
【问题描述】:
我尝试使用 CNTK 和 TF 在虚拟数据集上实现 char-LSTM 分类模型,这两个模型在完美数据集上都实现了 100% 的准确度(我现在不担心过度拟合)。
但是,一旦完成训练并将模型保存到文件并恢复,模型似乎忘记了训练,并且在同一数据集上表现不佳。
我比较了 LSTM 和密集层在训练后和恢复后的权重和偏差,它们完美匹配。我觉得还有一些东西需要恢复(可能是 LSTM 状态?),模型才能再次开始准确运行。
恢复的 LSTM 模型是否需要以某种方式“启动”才能再次开始满负荷运行?
代码和数据集可用here。
【问题讨论】:
标签: python tensorflow cntk