| """日志：tqdm / tensorboard""" | |
| from tqdm import tqdm | |
| class ProgressBar: | |
| """进度条""" | |
| def __init__(self, total, desc=""): | |
| self.pbar = tqdm(total=total, desc=desc) | |
| def update(self, n=1): | |
| """更新进度""" | |
| self.pbar.update(n) | |
| def close(self): | |
| """关闭进度条""" | |
| self.pbar.close() | |