【发布时间】：2021-09-04 10:18:28
【问题描述】：
我想用 MNIST 数据集训练 MLP（多层感知器）。我使用验证集，以便保存最佳模型的权重。然后我想将这些权重加载回相同的架构中，并使用它们来初始化和训练另一个数据集。我想知道 Tensorflow 1.x 或 2.x 是否可以做到这一点。现在我正在尝试编写一个自定义函数来做到这一点，但它变得越来越复杂。我正在使用 tf 1.x。
【问题讨论】：
标签： tensorflow keras neural-network
【发布时间】：2021-09-04 10:18:28
【问题描述】：
我建议你看看 tensorflow 的文档，这里有一个教程链接，可以保存你的权重并在之后加载它们：
【讨论】：