【从零学Python】一些工具类的使用：SummaryWriter()、tqdm()

1.SummaryWriter()
使用SummaryWriter创建events；
SummaryWriter()需要依赖tensorflow与tensorboard；

from torch.utils.tensorboard import SummaryWriter
log_dir = os.path.join(save_dir, 'models', datetime.now().strftime('%b%d_%H-%M-%S') + '_' + socket.gethostname())
# 获取Tensorboard的writer, 记录日志信息
writer = SummaryWriter(log_dir=log_dir)

上述代码执行后，会生成如下其中一个 run_* 文件

接下来在terminal中执行如下命令：

cd  run
tensorboard --logdir=run_2

点击下方链接：http://ubuntu-PT6630G-Invalid-entry-length-16-Fixed-up-to-11:6006
会得到如下界面：

2. tqdm()
   tqdm(),可扩展的Python进度条, https://www.jianshu.com/p/21cf48be6bf6

import time
from tqdm import tqdm

pbar = tqdm(["a", "b", "c", "d"])  
for char in pbar:  
    # 设置描述
    pbar.set_description("Processing %s" % char)
    time.sleep(1)

输出：

0%| | 0/4 [00:00

Processing a: 25%|██▌ | 1/4 [00:01<00:03, 1.00it/s]

Processing b: 50%|█████ | 2/4 [00:02<00:02, 1.00it/s]

Processing c: 75%|███████▌ | 3/4 [00:03<00:01, 1.00it/s]

Processing d: 100%|██████████| 4/4 [00:04<00:00, 1.00it/s]