深度学习——可视化界面（一） tensorboard

tensorboard 

CUDA_VISIBLE_DEVICES=4 tensorboard --logdir=/model/bert_model

4 是显卡位置， 显卡位置可以跟模型训练的不一致，而且可以在跑完模型之后查看    

 logdir 后面是模型路径， 注意模型路径不要加引号

命令行直接在shell窗口打

 

代码部分

estimator = tf.estimator.Estimator(model_fn, params=params, config=run_config)


run_config = tf.estimator.RunConfig(
        model_dir=checkpoint_dir,
        save_summary_steps=1,
        save_checkpoints_steps=500,
        session_config=session_config
    )

save_summary_steps 这里设置loss保存的距离，否则设置成500的话，就500个step一个loss统计点 

hook_dict = {}
hook_dict['loss'] = total_loss
hook_dict['global_steps'] = tf.train.get_or_create_global_step()
logging_hook = tf.train.LoggingTensorHook(
hook_dict, every_n_iter=args.save_summary_steps)