[PyTorch] 使用 Keras 風格進度條的套件： pkbar

Last Updated on 2020-12-22 by Clay

前言

pkbar 是最近在閱讀他人對於 pointer-generator network 實現時意外看到的一種『進度條』實現方法，號稱可以在 PyTorch 上使用 Keras 風格的進度條。 的確，Keras 預設的進度條還滿淺顯易懂，比起需要自己去印出進度條的 PyTorch 來說使用上更加快速。

當然，其實不僅是可以在 PyTorch 框架的程式碼中使用，也可以在任何迴圈中使用。

pkbar

第一次使用的時候，需要使用以下指令安裝：

pip3 install pkbar

然後，我們可以在任何時候啟用 pkbar 的進度條。

# coding: utf-8
import time
import pkbar


def main():
    iterate = 10
    bar = pkbar.Pbar(name="Progress", target=iterate)

    for i in range(iterate):
        time.sleep(0.3)
        bar.update(i)


if __name__ == '__main__':
    main()

Output:

Progress
10/10  [==============================] - 3.0s

References

• https://pypi.org/project/pkbar/
• https://towardsdatascience.com/recreating-keras-functional-api-with-pytorch-cc2974f7143c

Read More

• Python 中進程池 Pool 的使用方法及進度條顯示
• Linux 複製資料時顯示進度條
• Python 使用 tqdm 展示進度條，並嘗試自己寫進度條
• PyQt5 基本教學 (5) 進度條、滑動條、旋轉鈕