我想要控制 "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
2条答案
按热度按时间3lxsmp7m1#
我在 TensorFlow 1.14 中遇到了这个问题。即使在手动将钩子添加到训练函数中,它也会消失。是不是 TPU 的魔法把它消除了?
ttisahbt2#
这个问题有任何解决方案吗?