2

我有一个解码器函数,其中包含一个子函数,该函数在输入 2D 张量和 2xn 形变量上运行迭代求和。此求和函数在训练期间正常工作,但是函数中使用的变量 - 2xn 形变量和可能的偏差项 - 没有保存在任何模型检查点中。因此,当我加载一个模型进行解码时,函数失败,因为变量不在检查点中。我尝试将函数移出并将其用作普通函数,但仍然没有看到检查点中的变量。

我怀疑这可能是 saver.save() 中的默认保存范围有问题

4

0 回答 0