bert 在`tf.contrib.tpu.RunConfig`中,`log_step_count_steps`集合不起作用,

8cdiaqws  于 4个月前  发布在  其他
关注(0)|答案(2)|浏览(58)

我想要控制 "examples/sec" 信息消息的显示。因为我在谷歌Colab上训练模型,信息越多,浏览器就越卡。所以我设置了 log_step_count_steps 参数。但是当我修改代码如下时,它仍然输出:'_log_step_count_steps': None

run_config = tf.contrib.tpu.RunConfig(
        cluster=tpu_cluster_resolver,
        master=FLAGS.master,
        model_dir=FLAGS.output_dir,
        save_checkpoints_steps=FLAGS.save_checkpoints_steps,
        log_step_count_steps = FLAGS.log_step_count_steps,
        tpu_config=tf.contrib.tpu.TPUConfig(
            iterations_per_loop=FLAGS.iterations_per_loop,
            num_shards=FLAGS.num_tpu_cores,
            per_host_input_for_training=is_per_host))

。有人能帮我吗?非常感谢任何帮助。Tensorflow版本是1.12.0。


# 修改后的代码

log_step_count_steps = 100
3lxsmp7m

3lxsmp7m1#

我在 TensorFlow 1.14 中遇到了这个问题。即使在手动将钩子添加到训练函数中,它也会消失。是不是 TPU 的魔法把它消除了?

ttisahbt

ttisahbt2#

这个问题有任何解决方案吗?

相关问题