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

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'):

你可能感兴趣的:(深度学习)