当前位置: 代码迷 >> 综合 >> Tensorboard 监控模型训练过程按批次输出结果
  详细解决方案

Tensorboard 监控模型训练过程按批次输出结果

热度:28   发布时间:2024-01-28 16:09:07.0

Tensorboard监控模型训练过程时输出每个批次的结果

tensorboard = TensorBoard(log_dir="./logs",update_freq='batch')

要想以每个批次的结果输出,需要设置update_freq参数为batch,查看源码可以看到默认为epoch,所以如果不设置这个参数,每一轮结束后才会输出一次,很难实时监控模型训练状态。

    def __init__(self, log_dir='./logs',histogram_freq=0,batch_size=32,write_graph=True,write_grads=False,write_images=False,embeddings_freq=0,embeddings_layer_names=None,embeddings_metadata=None,embeddings_data=None,update_freq='epoch'):
  相关解决方案