欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

jupyter notebook不显示进度条progress.bar

程序员文章站 2022-01-20 10:06:53
...

为了方便jupyter notebook中显示训练、测试进度,之前的progress.bar因为不是tqdm写的,所以无法通过插件来转换显示。
解决:换成 print(’\r’, end="")就好了。

示例:
转换之前的代码:

        # for every batch
        bar.suffix = 'Batch ({b}/{bn}) ' \
                     '| Acc {acc:.4f} | BAcc {bacc:.4f} | Loss {loss_avg:.4f} | Total {total} | ETA {eta}'.format(
            b=i + 1,
            bn=len(train_loader),
            acc=acc_avg,
            bacc=bacc_avg,
            loss_avg=loss_avg,
            total=bar.elapsed_td,
            eta=bar.eta_td,
        )
        bar.next()
    bar.finish()

转换之后的代码:

        # for every batch
        print('\rBatch ({b}/{bn}) | Acc {acc:.4f} | BAcc {bacc:.4f} | Loss {loss_avg:.4f} | Total {total} | ETA {eta}'.format(
            b=i + 1,
            bn=len(train_loader),
            acc=acc_avg,
            bacc=bacc_avg,
            loss_avg=loss_avg,
            total=bar.elapsed_td,
            eta=bar.eta_td,
        ), end="")

参考:Python2/python3 print刷新打印(打印输出自动更新不换行)