openmmlab系列可视化训练过程

在 OpenMMLab 中 使用 TensorBoard 只需修改一行配置：

1. 安装 TensorBoard

pip install tensorboard

2. 修改 config 中的 log_config 字段（一般在configs/base/default_runtime.py）

log_config = dict( 
    interval=10, 
    hooks=[ 
        dict(type='TextLoggerHook'), 
        dict(type='TensorboardLoggerHook') 
    ])

3. 运行命令

python tools/train.py configs/resnet/resnet18_b16x8_cifar10.py #替换为你的
# 另起一个终端 
tensorboard --logdir work_dirs/swin#修改为你的文件夹

4.打开 chrome 浏览器 输入 http://localhost:6006/
如果想分享 tensorboard 日志，可以在上面的 tensorboard 命令中加入 dev 参数