【发布时间】:2019-10-02 21:28:19
【问题描述】:
禁用 Eager Execution 后,我可以连接到我的云 TPU 并运行我的自定义训练循环。计算损失后,我想打印该变量。
鉴于损失是 Cloud TPU 的张量,到目前为止我还没有找到任何方法来打印它。 tf.print 返回PrintOperation,所以我被卡住了。我想像将损失张量移回我的 cpu 之类的东西会起作用,但我还没有找到任何 hacky 方法来做到这一点。
我知道有可能获得该结果,因为当我在 TPU 分配策略下使用 keras 封装我的模型并使用 model.fit 而不是我的自定义训练循环时,我可以打印损失指标。
所以必须有一种方法,非常感谢任何帮助找到它:)
【问题讨论】:
标签: tensorflow keras cloud tpu