【从零学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
会得到如下界面:
- 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<?, ?it/s]
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]