python - 如何在 TensorFlow v2 中传递会话对象？

Question

我有一个函数change_weight()可以修改任何给定模型中的权重。此函数驻留在不同的 python 文件中。

因此，如果我有一个对 MNIST 图像进行分类的简单神经网络，我会在调用此函数之前和之后测试准确性，我发现它有效。这在 TensorFlow v1 中很容易做到，因为我只需要sess在函数调用中传递 Session 对象，就可以在另一个文件中获得该会话的权重。

在 TensorFlow v2 中使用 Eager Execution，我该怎么做？我没有 Session 对象了。我要通过什么？

Answer 1

我可以通过传递Model对象并model.trainable_variables在另一个函数中获取权重来做到这一点。