为了方便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="")