是否可以仅保存张量流模型的训练权重？

Question

我在云计算机上训练了一个模型，现在我想把它移植到Jetson TX2。看起来它无法从张量流的保护模块加载模型 可以手动创建图形，然后只加载权重？
如果是这样，我如何只保存模型的权重？

这不仅仅是一次保存和恢复，因为我已经尝试过这已经没有运气了。

Answer 1

正如@jdehesa评论的解决方案是冻结图表。 我做了冻结并移除了设备，我可以在Jetson板上加载模型。