jupyter notebook不显示进度条progress.bar

为了方便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刷新打印(打印输出自动更新不换行)

猜你喜欢

转载自blog.csdn.net/qxqxqzzz/article/details/107344818