【问题标题】:Is it possible to print value of TPU cloud variable in tensorflow?是否可以在 tensorflow 中打印 TPU 云变量的值?
【发布时间】:2019-10-02 21:28:19
【问题描述】:

禁用 Eager Execution 后,我可以连接到我的云 TPU 并运行我的自定义训练循环。计算损失后,我想打印该变量。

鉴于损失是 Cloud TPU 的张量,到目前为止我还没有找到任何方法来打印它。 tf.print 返回PrintOperation,所以我被卡住了。我想像将损失张量移回我的 cpu 之类的东西会起作用,但我还没有找到任何 hacky 方法来做到这一点。

我知道有可能获得该结果,因为当我在 TPU 分配策略下使用 keras 封装我的模型并使用 model.fit 而不是我的自定义训练循环时,我可以打印损失指标。

所以必须有一种方法,非常感谢任何帮助找到它:)

【问题讨论】:

    标签: tensorflow keras cloud tpu


    【解决方案1】:

    好的,找到我的 hacky 方法,即使存在更好的替代方案,我也很想听听:

    sess = tf.compat.v1.Session()
    init_op = tf.compat.v1.global_variables_initializer()
    sess.run(init_op)
    <tensor>.eval(session=sess)
    

    【讨论】:

    • 你有没有找到合适的方法?
    • @SantoshGupta7 哈哈,没有。
    • 我刚刚收到错误:“RuntimeError: 会话图为空。在调用 run() 之前向图中添加操作。”
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2013-04-19
    • 1970-01-01
    • 2010-11-15
    • 2021-02-14
    • 2017-11-10
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多